Searched defs:prim_node (Results 1 – 14 of 14) sorted by relevance
23 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
112 auto prim_node = cnode->input(0); in InsertPreTransNode() local
77 const auto &prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelConsiderCNode() local
47 auto prim_node = conv_cnode->input(0); in Run() local
278 auto prim_node = cnode->input(0); in InsertPreTransNode() local770 auto prim_node = cnode->input(0); in RunDoFixFormat() local
60 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsParallelCareNode() local
184 ValueNodePtr prim_node = cnode->input(0)->cast<ValueNodePtr>(); in IsAutoParallelCareNode() local
91 auto prim_node = GetValueNode<PrimitivePtr>(anf_node); in SetAllReduceRecomputeFlag() local
138 auto prim_node = GetCNodePrimitive(node); in CheckRandomEffect() local
192 auto prim_node = utils::cast<AnfNodePtr>(values_pattern[0]); in UpdateEquivMap() local
137 auto prim_node = cnode->input(0); in GetTransNodeFormatType() local
48 for (auto &prim_node : END_NODE_BLACK_LIST) { in IsInEndNodeBlackList() local
913 auto prim_node = cnode->input(0); in KeepValueNodeDuplication() local