提交 56f60ee5 编写于 作者: Y Yu Yang

Update example

上级 5ee70938
...@@ -56,14 +56,14 @@ We should chagne register macros at the same time. In the current solution, ther ...@@ -56,14 +56,14 @@ We should chagne register macros at the same time. In the current solution, ther
The user interface should be The user interface should be
```cpp ```cpp
vector<OpDesc> SumOpGradMakerƒ(OpDesc) {...} vector<OpDesc> MinusOpGradMaker(OpDesc) {...}
REGISTER_OPERATOR(sum, SumOp, SumOpProtoAndCheckerMaker, SumOpGradMaker); REGISTER_OPERATOR(minus, MinusOp, MinusOpProtoAndCheckerMaker, SumOpGradMaker);
// Developers can still manually implement gradient operator. // Developers can still manually implement gradient operator.
REGISTER_OPERATOR(sum_grad, SumGradOp); REGISTER_OPERATOR(minus_grad, MinusGradOp);
``` ```
The interface of current `REGISTER_OP` macro could not be changed. In `REGISTER_OP`, it will invoke `REGISTER_OPERATOR` two times and generate GradOpDescMaker inside. The interface of current `REGISTER_OP` macro could not be changed. In `REGISTER_OP`, it will invoke `REGISTER_OPERATOR` two times and generate GradOpDescMaker inside.
```cpp ```cpp
REGISTER_OP(sum, SumOp, SumOpProtoAndCheckerMaker, sum_grad, SumGradOp); REGISTER_OP(minus, MinusOp, MinusOpProtoAndCheckerMaker, minus_grad, MinusGradOp);
``` ```
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册