Searched defs:prim_anf_node (Results 1 – 5 of 5) sorted by relevance
288 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshape() local328 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindReshapePreNodeStraCosts() local381 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindReshapeNextNodeStraCosts() local
71 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetMiniStepOpDoMirrorLabel() local85 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetAllReduceRecomputeFlag() local502 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in HasStrategy() local574 ValueNodePtr prim_anf_node = use_cnode->input(0)->cast<ValueNodePtr>(); in StepRedistribution() local697 ValueNodePtr prim_anf_node = use_cnode->input(0)->cast<ValueNodePtr>(); in StepSplitTensor() local968 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindParameter() local1014 ValueNodePtr prim_anf_node = use_apply->input(0)->cast<ValueNodePtr>(); in FindCNode() local1813 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in FindPreNodes() local1863 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in CheckExtractInfomation() local1890 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ExtractInformation() local[all …]
412 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueId() local525 ValueNodePtr prim_anf_node = cnode->input(0)->cast<ValueNodePtr>(); in ConstructCostGraphNodesByUniqueIdTC() local660 ValueNodePtr prim_anf_node = inputs[0]->cast<ValueNodePtr>(); in ConstructCostGraphEdges() local1008 CNodePtr GetInternalOperatorInfo(const CNodePtr &cnode, const ValueNodePtr &prim_anf_node) { in GetInternalOperatorInfo()
168 auto prim_anf_node = new_node_input[0]->cast<ValueNodePtr>(); in SetCommunicationOpGroupLabel() local
296 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local335 auto prim_anf_node = node_input[0]->cast<ValueNodePtr>(); in operator() local