forked from mindspore-Ecosystem/mindspore
!16079 assgin add bug fix
From: @zyx5256 Reviewed-by: @wuxuejian,@liangchenghui Signed-off-by: @liangchenghui
This commit is contained in:
commit
053ec62018
|
@ -292,7 +292,7 @@ bool ArithmeticCPUKernel<T>::Launch(const std::vector<AddressPtr> &inputs, const
|
||||||
} else if (operate_type_ == POW) {
|
} else if (operate_type_ == POW) {
|
||||||
Pow(broadcastedInput1, broadcastedInput2, output);
|
Pow(broadcastedInput1, broadcastedInput2, output);
|
||||||
} else if (operate_type_ == ASSIGNADD) {
|
} else if (operate_type_ == ASSIGNADD) {
|
||||||
AssignAdd(broadcastedInput1, broadcastedInput2, output);
|
AssignAdd(input1, input2, output);
|
||||||
} else if (operate_type_ == ATAN2) {
|
} else if (operate_type_ == ATAN2) {
|
||||||
Atan2(broadcastedInput1, broadcastedInput2, output);
|
Atan2(broadcastedInput1, broadcastedInput2, output);
|
||||||
} else if (operate_type_ == SQUAREDDIFFERENCE) {
|
} else if (operate_type_ == SQUAREDDIFFERENCE) {
|
||||||
|
|
Loading…
Reference in New Issue