Searched defs:gather_cnode (Results 1 – 5 of 5) sorted by relevance
70 const auto &gather_cnode = gather_to_switch.first->cast<CNodePtr>(); in GetBranchNameToCondtionActor() local433 const auto &gather_cnode = input_with_index.first->cast<CNodePtr>(); in FixRefCountForRefNode() local490 const auto &gather_cnode = node->cast<CNodePtr>(); in FixRefCountForInputNode() local
187 auto gather_cnode = in CalResizeShape() local
88 auto gather_cnode = func_graph->NewCNode(all_gather_prim_c, {input}); in AdjustMegatronLinearAllGather() local
218 auto gather_cnode = resize_cnode->input(kInputIndexTwo)->cast<CNodePtr>(); in DoFuison() local
316 auto gather_cnode = opt::GenGatherNode(func_graph, gather_input, gather_indices, gather_name); in ConvertOnnxResizeForVariableShape() local