fix bug for int8 type input

5 years ago · e80d1d1921
parent 64ad8b235c
commit e80d1d1921
1 changed files with 1 additions and 1 deletions
--- 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) {