提交 ba1d4ed3 编写于 作者: 刘琦

Merge branch 'update_op_register_docs' into 'master'

update docs of op register

See merge request !704
...@@ -82,7 +82,37 @@ void Register_My_Custom_Op(OperatorRegistryBase *op_registry) { ...@@ -82,7 +82,37 @@ void Register_My_Custom_Op(OperatorRegistryBase *op_registry) {
} // namespace mace } // namespace mace
``` ```
And then register the new Op in `mace/core/operator.cc`. And then register the new Op in `mace/ops/ops_register.cc`.
```
#include "mace/ops/ops_register.h"
namespace mace {
namespace ops {
// Keep in lexicographical order
...
extern void Register_My_Custom_Op(OperatorRegistryBase *op_registry);
...
} // namespace ops
OperatorRegistry::OperatorRegistry() : OperatorRegistryBase() {
// Keep in lexicographical order
...
ops::Register_My_Custom_Op(this);
...
}
} // namespace mace
```
Implement the Op kernel code Implement the Op kernel code
---------------------------- ----------------------------
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册