| @@ -179,9 +179,9 @@ lite::Primitive *ModelImpl::CopyPrimitive(const schema::Primitive *src_prim) { | |||||
| case schema::PrimitiveType_StridedSlice: | case schema::PrimitiveType_StridedSlice: | ||||
| return new lite::StridedSlice(const_cast<schema::Primitive *>(src_prim)); | return new lite::StridedSlice(const_cast<schema::Primitive *>(src_prim)); | ||||
| case schema::PrimitiveType_Prelu: | case schema::PrimitiveType_Prelu: | ||||
| return new lite::Prelu(const_cast<schema::Primitive *>(srcPrim)); | |||||
| return new lite::Prelu(const_cast<schema::Primitive *>(src_prim)); | |||||
| case schema::PrimitiveType_CaffePReLU: | case schema::PrimitiveType_CaffePReLU: | ||||
| return new lite::CaffePReLU(const_cast<schema::Primitive *>(srcPrim)); | |||||
| return new lite::CaffePReLU(const_cast<schema::Primitive *>(src_prim)); | |||||
| case schema::PrimitiveType_Round: | case schema::PrimitiveType_Round: | ||||
| return new lite::Round(const_cast<schema::Primitive *>(src_prim)); | return new lite::Round(const_cast<schema::Primitive *>(src_prim)); | ||||
| case schema::PrimitiveType_Reverse: | case schema::PrimitiveType_Reverse: | ||||
| @@ -31,8 +31,7 @@ class PreluBaseCPUKernel : public LiteKernel { | |||||
| const std::vector<lite::tensor::Tensor *> &outputs, const Context *ctx, | const std::vector<lite::tensor::Tensor *> &outputs, const Context *ctx, | ||||
| const lite::Primitive *primitive) | const lite::Primitive *primitive) | ||||
| : LiteKernel(parameter, inputs, outputs, ctx, primitive), ctx_(ctx) { | : LiteKernel(parameter, inputs, outputs, ctx, primitive), ctx_(ctx) { | ||||
| opParameter->thread_num_ = ctx->thread_num_; | |||||
| prelu_param_ = reinterpret_cast<PreluParameter *>(opParameter); | |||||
| prelu_param_ = reinterpret_cast<PreluParameter *>(op_parameter_); | |||||
| } | } | ||||
| ~PreluBaseCPUKernel() = default; | ~PreluBaseCPUKernel() = default; | ||||
| @@ -51,12 +51,12 @@ int CaffePReluCPUKernel::Run() { | |||||
| MS_LOG(ERROR) << "Prepare fail!ret: " << prepare_ret; | MS_LOG(ERROR) << "Prepare fail!ret: " << prepare_ret; | ||||
| return 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(); | prelu_param_->input_num_ = input->ElementsNum(); | ||||
| input_data = reinterpret_cast<float *>(input->Data()); | 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(); | auto channels = input->shape(); | ||||
| prelu_param_->negtive_slope_ = reinterpret_cast<float *>(input1->Data()); | prelu_param_->negtive_slope_ = reinterpret_cast<float *>(input1->Data()); | ||||
| prelu_param_->channel_num_ = channels.at(channels.size() - 1); | prelu_param_->channel_num_ = channels.at(channels.size() - 1); | ||||
| @@ -32,7 +32,7 @@ class CaffePReluCPUKernel : public LiteKernel { | |||||
| const std::vector<lite::tensor::Tensor *> &outputs, const lite::Context *ctx, | const std::vector<lite::tensor::Tensor *> &outputs, const lite::Context *ctx, | ||||
| const lite::Primitive *primitive) | const lite::Primitive *primitive) | ||||
| : LiteKernel(parameter, inputs, outputs, ctx, primitive), ctx_(ctx), thread_count_(ctx->thread_num_) { | : LiteKernel(parameter, inputs, outputs, ctx, primitive), ctx_(ctx), thread_count_(ctx->thread_num_) { | ||||
| prelu_param_ = (reinterpret_cast<CaffePReluParameter *>(opParameter)); | |||||
| prelu_param_ = reinterpret_cast<CaffePReluParameter *>(op_parameter_); | |||||
| primitive_ = primitive; | primitive_ = primitive; | ||||
| } | } | ||||
| ~CaffePReluCPUKernel() = default; | ~CaffePReluCPUKernel() = default; | ||||