p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!7816 [MSLITE]int8 split bug: random output 

Merge pull request !7816 from ling/op
This commit is contained in:
mindspore-ci-bot 2020-10-27 20:37:42 +08:00 committed by Gitee
parent 81f610c528 d4c73e5683
commit b5577b92a8
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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