| @@ -532,6 +532,7 @@ table MatMul { | |||||
| table CaffePReLU { | table CaffePReLU { | ||||
| channelShared : bool = false; | channelShared : bool = false; | ||||
| slope: [float]; | |||||
| } | } | ||||
| table LeakyReLU { | table LeakyReLU { | ||||
| @@ -847,7 +848,7 @@ table OneHot { | |||||
| table Lstm{ | table Lstm{ | ||||
| bidirection: bool = false; | bidirection: bool = false; | ||||
| } | } | ||||
| table PriorBox { | table PriorBox { | ||||
| min_sizes: [int]; | min_sizes: [int]; | ||||
| max_sizes: [int]; | max_sizes: [int]; | ||||
| @@ -26,7 +26,7 @@ void CaffePRelu(float *input, float *output, CaffePReluParameter *prelu_param_, | |||||
| if (input[i] > 0) { | if (input[i] > 0) { | ||||
| output[i] = input[i]; | output[i] = input[i]; | ||||
| } else { | } else { | ||||
| if (prelu_param_->channeShared) { | |||||
| if (!prelu_param_->channeShared) { | |||||
| int temp = i % prelu_param_->channel_num_; | int temp = i % prelu_param_->channel_num_; | ||||
| output[i] = input[i] * prelu_param_->negtive_slope_[temp]; | output[i] = input[i] * prelu_param_->negtive_slope_[temp]; | ||||
| } else { | } else { | ||||