diff --git a/mace/ops/deconv_2d.cc b/mace/ops/deconv_2d.cc
index 9eba1bd014c8c21e835a2b4d72664b6a19769951..3779174d063d29c14e0c33bb101305d88392826a 100644
--- a/mace/ops/deconv_2d.cc
+++ b/mace/ops/deconv_2d.cc
@@ -270,6 +270,7 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) {
                   } else {
                     MACE_NOT_IMPLEMENTED;
                   }
+                  context->set_output_mem_type(mem_type);
                   FrameworkType framework_type =
                       static_cast<FrameworkType>(
                         ProtoArgHelper::GetOptionalArg<OperatorDef, int>(
@@ -279,8 +280,9 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) {
                     context->SetInputInfo(2, MemoryType::CPU_BUFFER,
                                           DataType::DT_INT32);
                   }
+                } else {
+                  context->set_output_mem_type(mem_type);
                 }
-                context->set_output_mem_type(mem_type);
               }));
 #endif  // MACE_ENABLE_OPENCL
 }