Browse Source

!9959 fix bug in insert format transform node pass

From: @hangangqiang
Reviewed-by: @ddwsky,@HilbertDavid
Signed-off-by: @ddwsky
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
c97cf843d9
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      mindspore/lite/tools/common/graph_util.cc

+ 4
- 0
mindspore/lite/tools/common/graph_util.cc View File

@@ -415,6 +415,7 @@ NodeIter InsertNodeBefore(schema::MetaGraphT *graphT, NodeIter existNodeIter, si
*errorCode = RET_NULL_PTR; *errorCode = RET_NULL_PTR;
return graphT->nodes.end(); return graphT->nodes.end();
} }
toAddTensor->nodeType = schema::NodeType_CNode;
preTensor->refCount = 0; preTensor->refCount = 0;
preTensor->data.clear(); preTensor->data.clear();
MS_ASSERT(toAddNodeIn->primitive != nullptr); MS_ASSERT(toAddNodeIn->primitive != nullptr);
@@ -456,6 +457,7 @@ NodeIter InsertNodeBefore(schema::MetaGraphT *graphT, NodeIter existNodeIter, si
MS_LOG(ERROR) << "Copy TensorT failed"; MS_LOG(ERROR) << "Copy TensorT failed";
return graphT->nodes.end(); return graphT->nodes.end();
} }
toAddTensor->nodeType = schema::NodeType_CNode;
MS_ASSERT(toAddNodeIn->primitive != nullptr); MS_ASSERT(toAddNodeIn->primitive != nullptr);
if (toAddNodeIn->primitive->value.type == schema::PrimitiveType_QuantDTypeCast) { if (toAddNodeIn->primitive->value.type == schema::PrimitiveType_QuantDTypeCast) {
auto prim = toAddNodeIn->primitive->value.AsQuantDTypeCast(); auto prim = toAddNodeIn->primitive->value.AsQuantDTypeCast();
@@ -515,6 +517,7 @@ NodeIter InsertNodeAfter(schema::MetaGraphT *graphT, NodeIter existNodeIter, siz
*errorCode = RET_NULL_PTR; *errorCode = RET_NULL_PTR;
return graphT->nodes.end(); return graphT->nodes.end();
} }
toAddTensor->nodeType = schema::NodeType_CNode;
MS_ASSERT(toAddNodeIn->primitive != nullptr); MS_ASSERT(toAddNodeIn->primitive != nullptr);
if (toAddNodeIn->primitive->value.type == schema::PrimitiveType_QuantDTypeCast) { if (toAddNodeIn->primitive->value.type == schema::PrimitiveType_QuantDTypeCast) {
auto prim = toAddNodeIn->primitive->value.AsQuantDTypeCast(); auto prim = toAddNodeIn->primitive->value.AsQuantDTypeCast();
@@ -559,6 +562,7 @@ NodeIter InsertNodeAfter(schema::MetaGraphT *graphT, NodeIter existNodeIter, siz
*errorCode = RET_NULL_PTR; *errorCode = RET_NULL_PTR;
return graphT->nodes.end(); return graphT->nodes.end();
} }
toAddTensor->nodeType = schema::NodeType_CNode;
graphT->allTensors.emplace_back(std::move(toAddTensor)); graphT->allTensors.emplace_back(std::move(toAddTensor));
size_t toAddTensorIdx = graphT->allTensors.size() - 1; size_t toAddTensorIdx = graphT->allTensors.size() - 1;
auto toAddNode = opDefCopyer(toAddNodeIn.get()); auto toAddNode = opDefCopyer(toAddNodeIn.get());


Loading…
Cancel
Save