| @@ -36,6 +36,9 @@ int SplitInt8CPUKernel::Init() { | |||||
| if (ret != RET_OK) { | if (ret != RET_OK) { | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| output_ptr_.resize(param->num_split_); | |||||
| auto in_tensor = in_tensors_.at(kInputIndex); | auto in_tensor = in_tensors_.at(kInputIndex); | ||||
| auto in_quant_args = in_tensor->GetQuantParams(); | auto in_quant_args = in_tensor->GetQuantParams(); | ||||
| @@ -90,7 +93,7 @@ int SplitInt8CPUKernel::Run() { | |||||
| input_ptr_ = reinterpret_cast<int8_t *>(in_tensor->MutableData()); | input_ptr_ = reinterpret_cast<int8_t *>(in_tensor->MutableData()); | ||||
| MS_ASSERT(param->num_split_ == outputs_.size()); | MS_ASSERT(param->num_split_ == outputs_.size()); | ||||
| for (int i = 0; i < param->num_split_; i++) { | for (int i = 0; i < param->num_split_; i++) { | ||||
| output_ptr_.push_back(reinterpret_cast<int8_t *>(out_tensors_.at(i)->MutableData())); | |||||
| output_ptr_[i] = reinterpret_cast<int8_t *>(out_tensors_.at(i)->data_c()); | |||||
| } | } | ||||
| auto ret = ParallelLaunch(this->context_->thread_pool_, SplitInt8Run, this, thread_n_num_); | auto ret = ParallelLaunch(this->context_->thread_pool_, SplitInt8Run, this, thread_n_num_); | ||||