Searched defs:prim_anf_node (Results 1 – 9 of 9) sorted by relevance
369 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshape() local429 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshapePreNodeStraCosts() local498 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindReshapeNextNodeStraCosts() local
107 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindPrevLayout() local533 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in CheckExtractInformation() local553 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in InitRefMap() local755 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ExtractGraphInformation() local
508 auto prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueId() local625 auto prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueIdTC() local800 ValueNodePtr prim_anf_node = inputs[0]->cast<ValueNodePtr>(); in ConstructCNodeCostGraphEdges() local1225 CNodePtr GetInternalOperatorInfo(const CNodePtr &cnode, const ValueNodePtr &prim_anf_node) { in GetInternalOperatorInfo()1440 auto prim_anf_node = GetValueNode<PrimitivePtr>(op->cnode()->input(0)); in HasUserConfiguredStrategy() local
53 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in GetCNodeOpName() local
87 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetAllReduceRecomputeFlag() local790 ValueNodePtr prim_anf_node = use_cnode->input(0)->cast<ValueNodePtr>(); in StepSplitTensor() local1933 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in CheckExtractInformation() local2023 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ExtractInformation() local2310 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindPrevLayout() local2351 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ReshapeInit() local2471 ValueNodePtr prim_anf_node = loss_cnode->input(0)->cast<ValueNodePtr>(); in GetLossNodeGradOutputLayout() local
943 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindParameter() local1605 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindParameterNextLayout() local
791 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetCommunicationOpGroupLabel() local2675 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindCNode() local
226 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local275 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local
281 auto prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in DealConv2dTransposeFusionNode() local