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 {