!16079 assgin add bug fix

From: @zyx5256
Reviewed-by: @wuxuejian,@liangchenghui
Signed-off-by: @liangchenghui
This commit is contained in:
mindspore-ci-bot 2021-05-08 09:34:18 +08:00 committed by Gitee
commit 053ec62018
1 changed files with 1 additions and 1 deletions

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