zhouqunjie
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!9959 fix bug in insert format transform node pass 

From: @hangangqiang
Reviewed-by: @ddwsky,@HilbertDavid
Signed-off-by: @ddwsky
This commit is contained in:
mindspore-ci-bot 2020-12-15 11:52:46 +08:00 committed by Gitee
parent c98e6b19d6 59ca96c1f2
commit c97cf843d9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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