!4044 Fix nasnet model impl

Merge pull request !4044 from ghzl/nasnet-model
pull/4044/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit c5173e44c9

@ -36,11 +36,7 @@ int SplitCPUKernel::Init() {
return RET_OK; return RET_OK;
} }
SplitBaseCPUKernel::Init(); SplitBaseCPUKernel::Init();
auto in_tensor = inputs_.front(); output_ptr_.resize(param->num_split_);
input_ptr_ = reinterpret_cast<float *>(in_tensor->Data());
for (int i = 0; i < param->num_split_; i++) {
output_ptr_.push_back(reinterpret_cast<float *>(outputs_.at(i)->Data()));
}
return RET_OK; return RET_OK;
} }
@ -73,6 +69,11 @@ int SplitRun(int task_id, LiteParallelGroupEnv *penv, void *cdata) {
int SplitCPUKernel::Run() { int SplitCPUKernel::Run() {
auto ret = Prepare(); auto ret = Prepare();
auto in_tensor = inputs_.front();
input_ptr_ = reinterpret_cast<float *>(in_tensor->Data());
for (int i = 0; i < output_ptr_.size(); i++) {
output_ptr_[i] = reinterpret_cast<float *>(outputs_.at(i)->Data());
}
if (ret != RET_OK) { if (ret != RET_OK) {
MS_LOG(ERROR) << "Prepare failed."; MS_LOG(ERROR) << "Prepare failed.";
return RET_ERROR; return RET_ERROR;

Loading…
Cancel
Save