提交 4223ae85 编写于 作者: 卢旭辉

Merge branch 'fix-deconv-set_memtype' into 'master'

fix deconv set mem type

See merge request !1177
......@@ -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
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册