Browse Source

assign add bug fix

pull/16079/head
zhuyuxiao 4 years ago
parent
commit
4e5a322f49
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc

+ 1
- 1
mindspore/ccsrc/backend/kernel_compiler/cpu/arithmetic_cpu_kernel.cc View File

@@ -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) {


Loading…
Cancel
Save