Searched defs:layout_ptr (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/allreduce_fusion/ |
| D | allreduce_node.cc | 86 auto layout_ptr = para_ptr->user_data<TensorLayout>(); in AddPara() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | reshape_layout_transfer.cc | 44 …std::shared_ptr<ReshapeLayoutTransfer> layout_ptr = std::make_shared<ReshapeLayoutTransfer>(*this); in UnifyDeviceArrangementAndTensorShape() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_assigned_parallel.cc | 76 auto layout_ptr = GetOutputLayoutFromCNode(cnode, output_index); in FindPrevParallelCareNodeLayout() local
101 auto layout_ptr = GetOutputLayoutFromCNode(cnode, 0); in FindPrevLayout() local
111 auto layout_ptr = FindPrevParallelCareNodeLayout(cnode->input(1), LongToSize(tuple_index)); in FindPrevLayout() local
123 auto layout_ptr = FindPrevLayout(cnode->inputs()[index]); in FindPrevLayout() local
176 auto layout_ptr = FindNextLayout(use_apply, next_is_reshape, -1); in FindNextLayout() local
|
| D | step_parallel.cc | 2136 …auto layout_ptr = FindNextLayout(use_apply, next_is_reshape, visit, make_tuple_index, tuple_get_in… in FindNextLayout() local
2180 auto layout_ptr = GetOutputLayoutFromCNode(cnode, output_index); in FindPrevParallelCareNodeLayout() local
2304 auto layout_ptr = GetOutputLayoutFromCNode(cnode, 0); in FindPrevLayout() local
2323 auto layout_ptr = FindPrevParallelCareNodeLayout(cnode->input(1), LongToSize(tuple_index)); in FindPrevLayout() local
2335 auto layout_ptr = FindPrevLayout(cnode->inputs()[index], is_input_param); in FindPrevLayout() local
|