Searched refs:prim_node (Results 1 – 14 of 14) sorted by relevance
77 const auto &prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelConsiderCNode() local78 if (prim_node == nullptr) { in IsParallelConsiderCNode()81 const auto &prim = prim_node->value()->cast<PrimitivePtr>(); in IsParallelConsiderCNode()
47 auto prim_node = conv_cnode->input(0); in Run() local48 MS_ASSERT(prim_node != nullptr); in Run()49 auto prim_value_node = prim_node->cast<ValueNodePtr>(); in Run()
278 auto prim_node = cnode->input(0); in InsertPreTransNode() local279 MS_CHECK_TRUE_MSG(prim_node != nullptr, lite::RET_ERROR, "prim_node is nullptr"); in InsertPreTransNode()280 auto prim = GetValueNode<PrimitivePtr>(prim_node); in InsertPreTransNode()770 auto prim_node = cnode->input(0); in RunDoFixFormat() local771 auto prim = GetValueNode<PrimitivePtr>(prim_node); in RunDoFixFormat()
23 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local24 auto prim = GetValueNode<PrimitivePtr>(prim_node); in GetTransNodeFormatType()
112 auto prim_node = cnode->input(0); in InsertPreTransNode() local113 auto prim = GetValueNode<PrimitivePtr>(prim_node); in InsertPreTransNode()
192 auto prim_node = utils::cast<AnfNodePtr>(values_pattern[0]); in UpdateEquivMap() local193 MS_EXCEPTION_IF_NULL(prim_node); in UpdateEquivMap()194 if (!IsValueNode<Primitive>(prim_node)) { in UpdateEquivMap()197 ValuePtr value = GetValueNode(prim_node); in UpdateEquivMap()
60 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelCareNode() local61 if (prim_node == nullptr) { in IsParallelCareNode()64 PrimitivePtr prim = prim_node->value()->cast<PrimitivePtr>(); in IsParallelCareNode()
184 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsAutoParallelCareNode() local185 if (prim_node == nullptr) { in IsAutoParallelCareNode()188 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_node); in IsAutoParallelCareNode()
91 auto prim_node = GetValueNode<PrimitivePtr>(anf_node); in SetAllReduceRecomputeFlag() local92 MS_EXCEPTION_IF_NULL(prim_node); in SetAllReduceRecomputeFlag()93 auto node_attrs = prim_node->attrs(); in SetAllReduceRecomputeFlag()97 …MS_LOG(INFO) << "Do not recompute the forward communication operator of " << prim_node->ToString(); in SetAllReduceRecomputeFlag()
138 auto prim_node = GetCNodePrimitive(node); in CheckRandomEffect() local148 if (prim_main->name() != prim_node->name()) { in CheckRandomEffect()
137 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local138 auto prim = GetValueNode<PrimitivePtr>(prim_node); in GetTransNodeFormatType()
913 auto prim_node = cnode->input(0); in KeepValueNodeDuplication() local914 if (IsValueNode<Primitive>(prim_node)) { in KeepValueNodeDuplication()915 auto prim = GetValue<PrimitivePtr>(prim_node->cast<ValueNodePtr>()->value()); in KeepValueNodeDuplication()
48 for (auto &prim_node : END_NODE_BLACK_LIST) { in IsInEndNodeBlackList() local49 if (IsPrimitiveCNode(cnode, prim_node)) { in IsInEndNodeBlackList()