Register assign_value_op an empty grad_op

732eef57 · fengjiayi · a29cb4be · 732eef57
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/operators/assign_value_op.cc paddle/fluid/operators/assign_value_op.cc +2 -1

未找到文件。
--- 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>);