From 91384f6955219d68bfbf806d612e32d12775ae9f Mon Sep 17 00:00:00 2001 From: zhengjun10 Date: Wed, 17 Mar 2021 16:48:37 +0800 Subject: [PATCH] fix tanhgrad bug --- .../lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc b/mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc index 155d71f1af..5d5e1390f6 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32_grad/activation_grad.cc @@ -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) {