p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!4382 Adding resize function for donzen of operators 

Merge pull request !4382 from zhangzheng/master
This commit is contained in:
mindspore-ci-bot 2020-08-14 18:30:20 +08:00 committed by Gitee
parent fe834fb15d ec2609cef4
commit 32208e367e
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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