diff --git a/paddle/operators/assign_value_op.cc b/paddle/operators/assign_value_op.cc index a0bce99ff3df4f9d921a0d40e3abd900b2f6f663..7f6d351f505278a2c887dac1bcefc9728b933b64 100644 --- a/paddle/operators/assign_value_op.cc +++ b/paddle/operators/assign_value_op.cc @@ -33,7 +33,7 @@ class AssignValueOp : public framework::OperatorWithKernel { } protected: - framework::OpKernelType GetActualKernelType( + framework::OpKernelType GetExpectedKernelType( const framework::ExecutionContext &ctx) const override { return framework::OpKernelType( framework::proto::DataType(ctx.Attr("dtype")), ctx.GetPlace());