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 { ...@@ -94,19 +94,13 @@ class OpRegistry {
const std::string& type, const VariableNameMap& inputs, const std::string& type, const VariableNameMap& inputs,
const VariableNameMap& outputs, const AttributeMap attrs, const VariableNameMap& outputs, const AttributeMap attrs,
std::shared_ptr<paddle_mobile::framework::Scope> scope) { std::shared_ptr<paddle_mobile::framework::Scope> scope) {
LOG(paddle_mobile::kLOG_DEBUG1) << " type: " LOG(paddle_mobile::kLOG_DEBUG1) << " type: " << type;
<< type; LOG(paddle_mobile::kLOG_DEBUG1) << " input size: " << inputs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " input size: " LOG(paddle_mobile::kLOG_DEBUG1) << " output size: " << outputs.size();
<< inputs.size(); LOG(paddle_mobile::kLOG_DEBUG1) << " attr size: " << attrs.size();
LOG(paddle_mobile::kLOG_DEBUG1) << " output size: " LOG(paddle_mobile::kLOG_DEBUG1)
<< outputs.size(); << " OpInfoMap size: " << OpInfoMap<Dtype>::Instance().map().size();
LOG(paddle_mobile::kLOG_DEBUG1) << " attr size: " LOG(paddle_mobile::kLOG_DEBUG1) << " has type: " << type << " "
<< 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); << OpInfoMap<Dtype>::Instance().Has(type);
auto& info = OpInfoMap<Dtype>::Instance().Get(type); auto& info = OpInfoMap<Dtype>::Instance().Get(type);
auto op = info.Creator()(type, inputs, outputs, attrs, scope); 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.
先完成此消息的编辑！
想要评论请 注册 或