Searched refs:prim_anf_node (Results 1 – 6 of 6) sorted by relevance
288 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshape() local289 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in FindReshape()328 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshapePreNodeStraCosts() local329 PrimitivePtr prim = prim_anf_node->value()->cast<PrimitivePtr>(); in FindReshapePreNodeStraCosts()381 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindReshapeNextNodeStraCosts() local382 MS_EXCEPTION_IF_NULL(prim_anf_node); in FindReshapeNextNodeStraCosts()383 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in FindReshapeNextNodeStraCosts()
71 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetMiniStepOpDoMirrorLabel() local72 auto prim = GetValueNode<PrimitivePtr>(prim_anf_node); in SetMiniStepOpDoMirrorLabel()85 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetAllReduceRecomputeFlag() local86 auto prim = GetValueNode<PrimitivePtr>(prim_anf_node); in SetAllReduceRecomputeFlag()502 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in HasStrategy() local503 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in HasStrategy()574 ValueNodePtr prim_anf_node = use_cnode->input(0)->cast<ValueNodePtr>(); in StepRedistribution() local575 MS_EXCEPTION_IF_NULL(prim_anf_node); in StepRedistribution()576 PrimitivePtr node_prim = prim_anf_node->value()->cast<PrimitivePtr>(); in StepRedistribution()697 ValueNodePtr prim_anf_node = use_cnode->input(0)->cast<ValueNodePtr>(); in StepSplitTensor() local[all …]
412 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueId() local416 auto prev_cnode = GetInternalOperatorInfo(cnode, prim_anf_node); in ConstructCostGraphNodesByUniqueId()423 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in ConstructCostGraphNodesByUniqueId()525 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueIdTC() local529 auto prev_cnode = GetInternalOperatorInfo(cnode, prim_anf_node); in ConstructCostGraphNodesByUniqueIdTC()536 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in ConstructCostGraphNodesByUniqueIdTC()660 ValueNodePtr prim_anf_node = inputs[0]->cast<ValueNodePtr>(); in ConstructCostGraphEdges() local664 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in ConstructCostGraphEdges()1008 CNodePtr GetInternalOperatorInfo(const CNodePtr &cnode, const ValueNodePtr &prim_anf_node) { in GetInternalOperatorInfo() argument1009 PrimitivePtr prim = GetValueNode<PrimitivePtr>(prim_anf_node); in GetInternalOperatorInfo()
168 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetCommunicationOpGroupLabel() local169 auto prim = GetValueNode<PrimitivePtr>(prim_anf_node); in SetCommunicationOpGroupLabel()
55 CNodePtr GetInternalOperatorInfo(const CNodePtr &cnode, const ValueNodePtr &prim_anf_node);
296 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local297 prim = GetValueNode<PrimitivePtr>(prim_anf_node); in operator()335 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local336 prim = GetValueNode<PrimitivePtr>(prim_anf_node); in operator()