| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | tensor_layout_test.cc | 41 Arrangement device_arrangement; in ReshapeExpandDeviceArrangementTestFunction() local
74 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
95 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
115 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
136 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
147 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
161 Arrangement device_arrangement; in ExpandTensorShapeTestFunction() local
195 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
206 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
220 Arrangement device_arrangement; in TEST_F() local
[all …]
|
| D | reshape_layout_transfer_test.cc | 43 Arrangement device_arrangement; in InferUnifiedLayout() local
88 void InferUnifiedLayoutCheck(const DeviceArrangement &device_arrangement, const TensorMap &in_tenso… in InferUnifiedLayoutCheck()
112 void ValidUnifiedLayoutCheck(const DeviceArrangement &device_arrangement, const TensorMap &in_tenso… in ValidUnifiedLayoutCheck()
131 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
146 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
161 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
170 DeviceArrangement device_arrangement = {8, 4}; in TEST_F() local
179 DeviceArrangement device_arrangement = {2}; in TEST_F() local
188 DeviceArrangement device_arrangement = {4, 8}; in TEST_F() local
197 DeviceArrangement device_arrangement = {2, 2}; in TEST_F() local
[all …]
|
| D | tensor_redistribution_test.cc | 53 DeviceArrangement device_arrangement = {2, 4, 2}; in TEST_F() local
106 DeviceArrangement device_arrangement = {16, 1, 1}; in TEST_F() local
158 DeviceArrangement device_arrangement = {8}; in TEST_F() local
|
| D | util_layout_gen_test.cc | 106 void GenerateValidTensorMap(const DeviceArrangement &device_arrangement, const TensorShape &tensor_… in GenerateValidTensorMap()
176 bool CheckLayoutValid(const DeviceArrangement &device_arrangement, const TensorMap &tensor_map, in CheckLayoutValid()
198 bool ShapeIsDividedByDevice(const DeviceArrangement &device_arrangement, const TensorMap &tensor_ma… in ShapeIsDividedByDevice()
236 void ComputeAccumDeviceTOAccumShapeMap(const DeviceArrangement &device_arrangement, const TensorMap… in ComputeAccumDeviceTOAccumShapeMap()
|
| D | construct_operator_test.cc | 84 Shape device_arrangement = {8, 4}; in TEST_F() local
|
| /third_party/mindspore/mindspore/parallel/ |
| D | _tensor.py | 42 def _get_tensor_slice_index(device_arrangement, tensor_strategy, tensor_map, rank_index): argument
61 def _rank_to_coordinate(rank_index, device_arrangement): argument
81 def _coordinate_to_rank(device_coordinate, device_arrangement): argument
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/graph_util/ |
| D | get_parallel_info.cc | 42 auto device_arrangement = tensor_layout->device_arrangement().array(); in GetParameterLayout() local
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | tensor_layout.cc | 48 Status TensorLayout::Init(const Arrangement &device_arrangement, const Map &tensor_map, in Init()
71 Status TensorLayout::InitFromVector(const Shape &device_arrangement, const Shape &tensor_map, in InitFromVector()
|
| D | tensor_layout.h | 61 Arrangement device_arrangement() const { return device_arrangement_; } in device_arrangement() function
|