From 836ab0da35dcea291fb1ff12e3d1e50d255bf0b6 Mon Sep 17 00:00:00 2001 From: yejianwu Date: Thu, 2 Aug 2018 17:25:31 +0800 Subject: [PATCH] update docs of op register --- docs/development/adding_a_new_op.md | 32 ++++++++++++++++++++++++++++- 1 file changed, 31 insertions(+), 1 deletion(-) diff --git a/docs/development/adding_a_new_op.md b/docs/development/adding_a_new_op.md index eda612d3..1b1910db 100644 --- a/docs/development/adding_a_new_op.md +++ b/docs/development/adding_a_new_op.md @@ -82,7 +82,37 @@ void Register_My_Custom_Op(OperatorRegistryBase *op_registry) { } // 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 ---------------------------- -- GitLab