diff --git a/doc/design/register_grad_op.md b/doc/design/register_grad_op.md index a793cfe96bd1447759c6665bb4573a030a30e54c..02c4d554d0747c3aa47e71771cbd00d054e8bce7 100644 --- a/doc/design/register_grad_op.md +++ b/doc/design/register_grad_op.md @@ -56,14 +56,14 @@ We should chagne register macros at the same time. In the current solution, ther The user interface should be ```cpp -vector SumOpGradMakerÆ’(OpDesc) {...} -REGISTER_OPERATOR(sum, SumOp, SumOpProtoAndCheckerMaker, SumOpGradMaker); +vector MinusOpGradMaker(OpDesc) {...} +REGISTER_OPERATOR(minus, MinusOp, MinusOpProtoAndCheckerMaker, SumOpGradMaker); // 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. ```cpp -REGISTER_OP(sum, SumOp, SumOpProtoAndCheckerMaker, sum_grad, SumGradOp); +REGISTER_OP(minus, MinusOp, MinusOpProtoAndCheckerMaker, minus_grad, MinusGradOp); ```