Adding Resize function for donzes of operators

pull/4382/head
zhangz0911gm 5 years ago
parent 05b03fe017
commit ec2609cef4

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

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