Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/enhancer/
Dsplit_inputs_for_reduce_scatter.cc78 auto reduce_scatter = NewCNode(reduce_scatter_inputs, func_graph); in RearrangeInputsForReduceScatter() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/ir_fusion/
Dmc2_fusion.cc174 auto reduce_scatter = VectorRef({is_reduce_scatter, matmul_x_w}); in DefineFusionPattern() local