Browse Source

!4382 Adding resize function for donzen of operators

Merge pull request !4382 from zhangzheng/master
tags/v0.7.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
32208e367e
2 changed files with 3 additions and 2 deletions
  1. +2
    -1
      mindspore/lite/schema/ops.fbs
  2. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c

+ 2
- 1
mindspore/lite/schema/ops.fbs View File

@@ -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];


+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/nnacl/caffeprelu.c View File

@@ -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 {


Loading…
Cancel
Save