提交 803b7b62 编写于 作者: F fengjiayi

Follow comments

上级 097f533b
...@@ -54,6 +54,8 @@ struct OpInfo { ...@@ -54,6 +54,8 @@ struct OpInfo {
"Operator GradOpMaker has not been registered."); "Operator GradOpMaker has not been registered.");
return grad_op_maker_; return grad_op_maker_;
} }
const OpAttrChecker* Checker() const { return checker_; }
}; };
class OpInfoMap { class OpInfoMap {
......
...@@ -23,8 +23,8 @@ std::unique_ptr<OperatorBase> OpRegistry::CreateOp( ...@@ -23,8 +23,8 @@ std::unique_ptr<OperatorBase> OpRegistry::CreateOp(
const std::string& type, const VariableNameMap& inputs, const std::string& type, const VariableNameMap& inputs,
const VariableNameMap& outputs, AttributeMap attrs) { const VariableNameMap& outputs, AttributeMap attrs) {
auto& info = OpInfoMap::Instance().Get(type); auto& info = OpInfoMap::Instance().Get(type);
if (info.checker_ != nullptr) { if (info.Checker() != nullptr) {
info.checker_->Check(attrs); info.Checker()->Check(attrs);
} }
auto op = info.Creator()(type, inputs, outputs, attrs); auto op = info.Creator()(type, inputs, outputs, attrs);
return std::unique_ptr<OperatorBase>(op); return std::unique_ptr<OperatorBase>(op);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册