!6596 [MS][LITE]fix circumvention code

Merge pull request !6596 from gongdaguo/fix_bug
5 years ago · d8099576d3
parent cb97cd1b92 d9484235cb
commit d8099576d3
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
@ -158,7 +158,7 @@ STATUS DTypeTransPass::DoNodeInoutDTypeTrans(schema::MetaGraphT *graph) {
      if (!preTensor->data.empty() && !IsContain(graphInIdxes, (*iter)->inputIndex.at(i))) {
        continue;
      }
-      if (IsContain(graphInIdxes, (*iter)->inputIndex.at(i))) {
+      if ((preTensor->dataType != TypeId::kNumberTypeInt8) && (IsContain(graphInIdxes, (*iter)->inputIndex.at(i)))) {
        continue;
      }
      iter = InsertDTypeTransNode(graph, iter, kBefore, i, kInt8ToFP32, &status);