Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dsplit_inputs_for_reduce_scatter.cc38 std::vector<size_t> output_node_shape = AnfAlgo::GetPrevNodeOutputInferShape(node, i); in InsertSplitForInput() local
39 output_node_shape[0] /= rank_size_t; in InsertSplitForInput()
40 shapes.push_back(output_node_shape); in InsertSplitForInput()
41 size_splits.push_back(output_node_shape[0]); in InsertSplitForInput()