Home
last modified time | relevance | path

Searched defs:all_gather_cnode (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Doptimize_parallel_allgather_comm.cc38 void MoveCastBehindAllGather(const FuncGraphPtr &func_graph, const CNodePtr &all_gather_cnode, in MoveCastBehindAllGather()
113 auto all_gather_cnode = node->cast<CNodePtr>(); in OptimizeParallelAllGatherComm() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dmc2_fusion.cc260 auto all_gather_cnode = matmul_cnode->input(kIndex1)->cast<CNodePtr>(); in CreateFusionCNode() local