|
|
|
@ -70,7 +70,7 @@ int PReluCPUKernel::ProcessInput() {
|
|
|
|
|
MS_LOG(ERROR) << "malloc input_data_ failed.";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
memcpy(input_data_, ori_input_, tile_block * TILE_NUM * channel_num * sizeof(float));
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
|
|
|
|
|
return RET_OK;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
@ -85,7 +85,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
|
|
|
|
|
MS_LOG(ERROR) << "malloc input_data_ failed.";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 64 * sizeof(float));
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
|
|
|
|
|
#elif ENABLE_ARM32
|
|
|
|
|
prelu_param_->tile_block_ = UP_DIV(prelu_param_->input_num_, 32);
|
|
|
|
|
input_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(prelu_param_->tile_block_ * 32 * sizeof(float)));
|
|
|
|
@ -93,7 +93,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
|
|
|
|
|
MS_LOG(ERROR) << "malloc input_data_ failed.";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 32 * sizeof(float));
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
|
|
|
|
|
#else
|
|
|
|
|
prelu_param_->tile_block_ = UP_DIV(prelu_param_->input_num_, 32);
|
|
|
|
|
input_data_ = reinterpret_cast<float *>(context_->allocator->Malloc(prelu_param_->tile_block_ * 32 * sizeof(float)));
|
|
|
|
@ -101,7 +101,7 @@ int PReluCPUKernel::ProcessShareChannelInput() {
|
|
|
|
|
MS_LOG(ERROR) << "malloc input_data_ failed.";
|
|
|
|
|
return RET_ERROR;
|
|
|
|
|
}
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->tile_block_ * 32 * sizeof(float));
|
|
|
|
|
memcpy(input_data_, ori_input_, prelu_param_->input_num_ * sizeof(float));
|
|
|
|
|
#endif
|
|
|
|
|
return RET_OK;
|
|
|
|
|
}
|
|
|
|
|