diff --git a/mace/ops/deconv_2d.h b/mace/ops/deconv_2d.h index d4bdc11e2c90f80e11f3f40102481919e5ddfc51..03c4581d4af177d7918f41094668c3860ae74a3e 100644 --- a/mace/ops/deconv_2d.h +++ b/mace/ops/deconv_2d.h @@ -45,7 +45,7 @@ class Deconv2dOp : public Operator { MACE_CHECK(this->InputSize() >= 2, "deconv needs >= 2 inputs."); const Tensor *input = this->Input(0); const Tensor *filter = this->Input(1); - const kernels::FrameworkType model_type = + kernels::FrameworkType model_type = static_cast( OperatorBase::GetOptionalArg("framework_type", 0)); if (model_type == kernels::CAFFE) {