Skip to content

P
Paddle-Lite

PaddlePaddle / Paddle-Lite

通知 338
Star 4
Fork 1
P
Paddle-Lite
提交 84dcb100 编写于 作者: Z zhaojiaying01
浏览文件
操作

op register

上级 a39deed4
显示空白变更内容
内联 并排
Showing with 7 addition and 13 deletion
src/framework/op_registry.h
浏览文件 @ 84dcb100
......@@ -94,19 +94,13 @@ class OpRegistry {
const std::string& type, const VariableNameMap& inputs,
const VariableNameMap& outputs, const AttributeMap attrs,
std::shared_ptr<paddle_mobile::framework::Scope> scope) {
LOG(paddle_mobile::kLOG_DEBUG1) << " type: "
<< type;
LOG(paddle_mobile::kLOG_DEBUG1) << " input size: "
<< inputs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " output size: "
<< outputs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " attr size: "
<< attrs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " OpInfoMap size: "
<< OpInfoMap<Dtype>::Instance().map().size();
LOG(paddle_mobile::kLOG_DEBUG1) << " has type: "
<< type
<< " "
LOG(paddle_mobile::kLOG_DEBUG1) << " type: " << type;
LOG(paddle_mobile::kLOG_DEBUG1) << " input size: " << inputs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " output size: " << outputs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " attr size: " << attrs.size();
LOG(paddle_mobile::kLOG_DEBUG1)
<< " OpInfoMap size: " << OpInfoMap<Dtype>::Instance().map().size();
LOG(paddle_mobile::kLOG_DEBUG1) << " has type: " << type << " "
<< OpInfoMap<Dtype>::Instance().Has(type);
auto& info = OpInfoMap<Dtype>::Instance().Get(type);
auto op = info.Creator()(type, inputs, outputs, attrs, scope);
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或