diff --git a/paddle/operators/assign_value_op.cc b/paddle/operators/assign_value_op.cc
index a0bce99ff3..7f6d351f50 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<int>("dtype")), ctx.GetPlace());