From ec2609cef49c8d8446ca632f21d5f477506567cc Mon Sep 17 00:00:00 2001 From: zhangz0911gm Date: Wed, 12 Aug 2020 03:52:08 -0400 Subject: [PATCH] Adding Resize function for donzes of operators --- mindspore/lite/schema/ops.fbs | 3 ++- mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/lite/schema/ops.fbs b/mindspore/lite/schema/ops.fbs index cbb07b0be8..a8ee4858ce 100644 --- a/mindspore/lite/schema/ops.fbs +++ b/mindspore/lite/schema/ops.fbs @@ -532,6 +532,7 @@ table MatMul { table CaffePReLU { channelShared : bool = false; + slope: [float]; } table LeakyReLU { @@ -847,7 +848,7 @@ table OneHot { table Lstm{ bidirection: bool = false; } - + table PriorBox { min_sizes: [int]; max_sizes: [int]; diff --git a/mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c b/mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c index 76704d287f..4eb5943eec 100644 --- a/mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c +++ b/mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c @@ -26,7 +26,7 @@ void CaffePRelu(float *input, float *output, CaffePReluParameter *prelu_param_, if (input[i] > 0) { output[i] = input[i]; } else { - if (prelu_param_->channeShared) { + if (!prelu_param_->channeShared) { int temp = i % prelu_param_->channel_num_; output[i] = input[i] * prelu_param_->negtive_slope_[temp]; } else {