Home
last modified time | relevance | path

Searched refs:reduce_scatter_inputs (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc63 std::vector<AnfNodePtr> reduce_scatter_inputs{ in RearrangeInputsForReduceScatter() local
68 reduce_scatter_inputs.push_back(inputs[idx]); in RearrangeInputsForReduceScatter()
71 auto reduce_scatter = func_graph->NewCNode(reduce_scatter_inputs); in RearrangeInputsForReduceScatter()