!5338 fix bug for int8 input

Merge pull request !5338 from cjh9368/aware_quant

!5338 fix bug for int8 input
Merge pull request !5338 from cjh9368/aware_quant
acafe51a · mindspore-ci-bot · Gitee · 8e80f80f · e80d1d19 · acafe51a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/lite/tools/converter/legacy_optimizer/graph/dtype_trans_pass.cc ...ools/converter/legacy_optimizer/graph/dtype_trans_pass.cc +1 -1

未找到文件。
--- a/mindspore/lite/tools/converter/legacy_optimizer/graph/dtype_trans_pass.cc
+++ b/mindspore/lite/tools/converter/legacy_optimizer/graph/dtype_trans_pass.cc
@@ -56,7 +56,7 @@ STATUS DTypeTransPass::DoModelInputDTypeTrans(schema::MetaGraphT *graph) {
  for (auto graphInIdx : graphInIdxes) {
    MS_ASSERT(graph->allTensors.size() > graphInIdx);
    auto &graphInTensor = graph->allTensors.at(graphInIdx);
-    graphInTensor->dataType = TypeId::kNumberTypeUInt8;
+    graphInTensor->dataType = TypeId::kNumberTypeInt8;
  }

  if (this->inputDataDType == TypeId::kNumberTypeInt8) {