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

fix deconv set mem type

上级 708aa21e
......@@ -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.
先完成此消息的编辑!
想要评论请 注册