Browse Source

!6596 [MS][LITE]fix circumvention code

Merge pull request !6596 from gongdaguo/fix_bug
tags/v1.0.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
d8099576d3
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/tools/converter/legacy_optimizer/graph/dtype_trans_pass.cc

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

@@ -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);


Loading…
Cancel
Save