提交 048dbe96 编写于 作者: L liutuo

fix deconv set mem type

上级 708aa21e
...@@ -270,6 +270,7 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) { ...@@ -270,6 +270,7 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) {
} else { } else {
MACE_NOT_IMPLEMENTED; MACE_NOT_IMPLEMENTED;
} }
context->set_output_mem_type(mem_type);
FrameworkType framework_type = FrameworkType framework_type =
static_cast<FrameworkType>( static_cast<FrameworkType>(
ProtoArgHelper::GetOptionalArg<OperatorDef, int>( ProtoArgHelper::GetOptionalArg<OperatorDef, int>(
...@@ -279,8 +280,9 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) { ...@@ -279,8 +280,9 @@ void RegisterDeconv2D(OpRegistryBase *op_registry) {
context->SetInputInfo(2, MemoryType::CPU_BUFFER, context->SetInputInfo(2, MemoryType::CPU_BUFFER,
DataType::DT_INT32); DataType::DT_INT32);
} }
} } else {
context->set_output_mem_type(mem_type); context->set_output_mem_type(mem_type);
}
})); }));
#endif // MACE_ENABLE_OPENCL #endif // MACE_ENABLE_OPENCL
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册