Browse Source

!11456 [MSLITE] fix tanhgrad issue bug

From: @zhengjun10
Reviewed-by: @zhang_xue_tong,@ivss,@ehaleva
Signed-off-by: @zhang_xue_tong
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 4 years ago
parent
commit
06107de3f7
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc

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


Loading…
Cancel
Save