Searched refs:IsApply (Results 1 – 8 of 8) sorted by relevance
45 …if (IsApply(prim::kPrimScalarSummary) || IsApply(prim::kPrimTensorSummary) || IsApply(prim::kPrimI… in fullname_with_scope()46 IsApply(prim::kPrimHistogramSummary)) { in fullname_with_scope()49 if (IsApply(prim::kPrimScalarSummary)) { in fullname_with_scope()51 } else if (IsApply(prim::kPrimImageSummary)) { in fullname_with_scope()53 } else if (IsApply(prim::kPrimHistogramSummary)) { in fullname_with_scope()
45 bool CNode::IsApply(const PrimitivePtr &value) const { in IsApply() function in mindspore::CNode167 return cnode->IsApply(value); in IsPrimitiveCNode()
265 bool IsApply(const PrimitivePtr &) const;
627 if (cnode->IsApply(std::make_shared<Primitive>("BiasAdd")) && in MatchAndMark()632 } else if (cnode->IsApply(std::make_shared<Primitive>("BiasAdd")) && in MatchAndMark()637 } else if (cnode->IsApply(prim::kPrimTupleGetItem) && in MatchAndMark()643 } else if (cnode->IsApply(prim::kPrimTupleGetItem) && in MatchAndMark()649 } else if (cnode->IsApply(prim::kPrimTupleGetItem) && in MatchAndMark()680 if (cnode->IsApply(prim::kPrimMakeTuple)) { in ExportNodes()686 if (nextCNode->IsApply(prim::kPrimUpdateState) && in ExportNodes()706 if (cnode->IsApply(prim::kPrimExpandDims)) { in ExportNodes()710 if (cnode->IsApply(prim::kPrimBatchMatMul)) { in ExportNodes()714 if (cnode->IsApply(prim::kPrimGeLU)) { in ExportNodes()[all …]
161 if (!cnode->IsApply(prim::kPrimDepend) && !cnode->IsApply(prim::kPrimUpdateState)) { in SearchTransDataAndCast()
66 assert(app.IsApply(primitive)); in TEST_F()
381 if (output_c->IsApply(prim::kPrimMakeTuple)) { in StructureOutput()390 if (output_c->IsApply(prim::kPrimDepend)) { in StructureOutput()
651 } else if (cnode->IsApply(prim::kPrimPartial) && !IsValueNode<Primitive>(inputs[1])) { in EraseTuple()