Browse Source

[MSLITE]int8 split bug

tags/v1.1.0
ling 5 years ago
parent
commit
d4c73e5683
1 changed files with 4 additions and 1 deletions
  1. +4
    -1
      mindspore/lite/src/runtime/kernel/arm/int8/split_int8.cc

+ 4
- 1
mindspore/lite/src/runtime/kernel/arm/int8/split_int8.cc View File

@@ -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_);


Loading…
Cancel
Save