diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc
index 333c9dbb4d4..60482e5c8a9 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc
@@ -292,7 +292,7 @@ bool ArithmeticCPUKernel<T>::Launch(const std::vector<AddressPtr> &inputs, const
   } else if (operate_type_ == POW) {
     Pow(broadcastedInput1, broadcastedInput2, output);
   } else if (operate_type_ == ASSIGNADD) {
-    AssignAdd(broadcastedInput1, broadcastedInput2, output);
+    AssignAdd(input1, input2, output);
   } else if (operate_type_ == ATAN2) {
     Atan2(broadcastedInput1, broadcastedInput2, output);
   } else if (operate_type_ == SQUAREDDIFFERENCE) {