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