Home
last modified time | relevance | path

Searched defs:micro_interleaved_index (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/pass/
Dfull_micro_interleaved_order_control.cc99 …size_t micro_interleaved_index = GetValue<size_t>(cnode->GetAttr(parallel::MICRO_INTERLEAVED_INDEX… in ExtractInterLeavedCommNode() local
131 void CreateGroupForMicroInterleaved(const CNodePtr &comm_cnode, size_t micro_interleaved_index) { in CreateGroupForMicroInterleaved()
Dlabel_micro_interleaved_index.cc45 …icroInterleavedIndexForForwardCommNodes(const CNodePtr &input_node, size_t micro_interleaved_index, in SpreadMicroInterleavedIndexForForwardCommNodes()
Dmicro_interleaved_order_control.cc132 …size_t micro_interleaved_index = GetValue<size_t>(cnode->GetAttr(parallel::MICRO_INTERLEAVED_INDEX… in ExtractInterLeavedCommNode() local
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/optimizer/
Dslice_activation_in_recompute.cc99 …auto micro_interleaved_index = GetValue<size_t>(allgather_cnode->GetAttr(parallel::MICRO_INTERLEAV… in CreateGroupForSliceAllGatherInMicroInterleaved() local
232 …auto micro_interleaved_index = GetValue<size_t>(allgather_node->GetAttr(parallel::MICRO_INTERLEAVE… in InsertAllGatherDependWithMicroInterleaved() local