Register assign_value_op an empty grad_op

7 years ago · 732eef57f5
parent a29cb4be2a
commit 732eef57f5
1 changed files with 2 additions and 1 deletions
--- a/paddle/fluid/operators/assign_value_op.cc
+++ b/paddle/fluid/operators/assign_value_op.cc
@ -70,6 +70,7 @@ $$Out = values$$

 namespace ops = paddle::operators;

-REGISTER_OPERATOR(assign_value, ops::AssignValueOp, ops::AssignValueOpMaker);
+REGISTER_OPERATOR(assign_value, ops::AssignValueOp, ops::AssignValueOpMaker,
+                  paddle::framework::EmptyGradOpMaker);
 REGISTER_OP_CPU_KERNEL(assign_value, ops::AssignValueKernel<int>,
                       ops::AssignValueKernel<float>);