Browse Source

Adding Resize function for donzes of operators

tags/v0.7.0-beta
zhangz0911gm 5 years ago
parent
commit
ec2609cef4
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