|
|
|
@ -51,12 +51,12 @@ int CaffePReluCPUKernel::Run() {
|
|
|
|
|
MS_LOG(ERROR) << "Prepare fail!ret: " << prepare_ret;
|
|
|
|
|
return prepare_ret;
|
|
|
|
|
}
|
|
|
|
|
auto input = inputs_.at(0);
|
|
|
|
|
auto input1 = inputs_.at(1);
|
|
|
|
|
auto input = in_tensors_[0];
|
|
|
|
|
auto input1 = in_tensors_[1];
|
|
|
|
|
|
|
|
|
|
prelu_param_->input_num_ = input->ElementsNum();
|
|
|
|
|
input_data = reinterpret_cast<float *>(input->Data());
|
|
|
|
|
output_data = reinterpret_cast<float *>(outputs_.at(0)->Data());
|
|
|
|
|
output_data = reinterpret_cast<float *>(out_tensors_[0]->Data());
|
|
|
|
|
auto channels = input->shape();
|
|
|
|
|
prelu_param_->negtive_slope_ = reinterpret_cast<float *>(input1->Data());
|
|
|
|
|
prelu_param_->channel_num_ = channels.at(channels.size() - 1);
|
|
|
|
|