!6596 [MS][LITE]fix circumvention code

Merge pull request !6596 from gongdaguo/fix_bug
This commit is contained in:
mindspore-ci-bot 2020-09-20 10:37:15 +08:00 committed by Gitee
commit d8099576d3
1 changed files with 1 additions and 1 deletions

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