|
|
|
@@ -36,6 +36,9 @@ int SplitInt8CPUKernel::Init() { |
|
|
|
if (ret != RET_OK) { |
|
|
|
return ret; |
|
|
|
} |
|
|
|
|
|
|
|
output_ptr_.resize(param->num_split_); |
|
|
|
|
|
|
|
auto in_tensor = in_tensors_.at(kInputIndex); |
|
|
|
|
|
|
|
auto in_quant_args = in_tensor->GetQuantParams(); |
|
|
|
@@ -90,7 +93,7 @@ int SplitInt8CPUKernel::Run() { |
|
|
|
input_ptr_ = reinterpret_cast<int8_t *>(in_tensor->MutableData()); |
|
|
|
MS_ASSERT(param->num_split_ == outputs_.size()); |
|
|
|
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_); |
|
|
|
|