xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!11456 [MSLITE] fix tanhgrad issue bug 

From: @zhengjun10
Reviewed-by: @zhang_xue_tong,@ivss,@ehaleva
Signed-off-by: @zhang_xue_tong
This commit is contained in:
mindspore-ci-bot 2021-03-18 09:28:02 +08:00 committed by Gitee
parent 9bac30d37f 91384f6955
commit 06107de3f7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc
View File

@ -66,7 +66,7 @@ int ActivationGradCPUKernel::DoActivation(int task_id) {
// Sigmoid gets the input tensors in reverse order!
error_code = SigmoidGrad(input_addr + start, yt_addr + start, count, output_addr + start);
} else if (param_act_grad_->type_ == schema::ActivationType_TANH) {
error_code = TanhGrad(yt_addr + start, input_addr + start, count, output_addr + start);
error_code = TanhGrad(input_addr + start, yt_addr + start, count, output_addr + start);
} else if (param_act_grad_->type_ == schema::ActivationType_HSWISH) {
error_code = HSwishGrad(yt_addr + start, input_addr + start, count, output_addr + start);
} else if (param_act_grad_->type_ == schema::ActivationType_HSIGMOID) {