From 265cb33093485b2481b2ec9e10a780f368ca67a4 Mon Sep 17 00:00:00 2001 From: liutuo Date: Wed, 17 Oct 2018 10:14:11 +0800 Subject: [PATCH] fix deconv op const var --- mace/ops/deconv_2d.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mace/ops/deconv_2d.h b/mace/ops/deconv_2d.h index d4bdc11e..03c4581d 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) { -- GitLab