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

fix quantiztion codegen for MCUs

This commit is contained in:
lz 2022-03-11 11:55:40 +08:00
parent 0b1ec574cf
commit ce35cb021d
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/lite/tools/converter/micro/coder/opcoders/base/quant_dtype_cast_coder.cc
View File

@ -41,9 +41,8 @@ int QuantDTypeCastCoder::DoCode(CoderContext *const context) {
MS_LOG(ERROR) << "QuantDTypeCast need quantization parameters which is not found.";
return RET_ERROR;
}
auto quant_arg = (!output_tensor_->quant_params().empty() && output_tensor_->quant_params().at(0).inited)
? output_tensor_->quant_params().at(0)
: input_tensor_->quant_params().at(0);
auto quant_arg = (output_tensor_->quant_params().empty()) ? input_tensor_->quant_params().at(0)
: output_tensor_->quant_params().at(0);
int num_unit_thread = input_tensor_->ElementsNum();
Collect(context,