Searched defs:in_tensor_map (Results 1 – 10 of 10) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | reshape_layout_transfer_test.cc | 46 Map in_tensor_map; in InferUnifiedLayout() local
88 …nferUnifiedLayoutCheck(const DeviceArrangement &device_arrangement, const TensorMap &in_tensor_map, in InferUnifiedLayoutCheck()
112 …alidUnifiedLayoutCheck(const DeviceArrangement &device_arrangement, const TensorMap &in_tensor_map, in ValidUnifiedLayoutCheck()
132 TensorMap in_tensor_map = {1, 0}; in TEST_F() local
147 TensorMap in_tensor_map = {1, 0}; in TEST_F() local
162 TensorMap in_tensor_map = {1, 0}; in TEST_F() local
171 TensorMap in_tensor_map = {1, -1}; in TEST_F() local
180 TensorMap in_tensor_map = {-1, -1}; in TEST_F() local
189 TensorMap in_tensor_map = {-1, -1, -1}; in TEST_F() local
198 TensorMap in_tensor_map = {0, -1}; in TEST_F() local
[all …]
|
| D | redistribution_layout_transfer_test.cc | 46 Map in_tensor_map; in RedistributionLayoutTransferTestFunction() local
89 …ributionLayoutCheck(const DeviceArrangement &in_device_arrangement, const TensorMap &in_tensor_map, in RedistributionLayoutCheck()
134 TensorMap in_tensor_map = {1, 0}; in TEST_F() local
169 TensorMap in_tensor_map = {1, 0}; in TEST_F() local
184 TensorMap in_tensor_map = {0, -1}; in TEST_F() local
199 TensorMap in_tensor_map = {2, 0}; in TEST_F() local
214 TensorMap in_tensor_map = {0, -1}; in TEST_F() local
227 …ributionLayoutCheck(const DeviceArrangement &in_device_arrangement, const TensorMap &in_tensor_map, in ValidRedistributionLayoutCheck()
253 TensorMap in_tensor_map = std::get<1>(layout_list[in]); in ValidRedistributionLayoutCheckAll() local
|
| D | tensor_redistribution_test.cc | 60 Map in_tensor_map; in TEST_F() local
113 Map in_tensor_map; in TEST_F() local
165 Map in_tensor_map; in TEST_F() local
|
| D | redistribution_operator_infer_test.cc | 52 void InferOperatorCheck(Shape in_tensor_map, const Shape &out_tensor_map, const OperatorList &opera… in InferOperatorCheck()
87 Map in_tensor_map; in InferOperatorCheckAll() local
|
| D | util_layout_gen_test.cc | 266 … const TensorMap &in_tensor_map, const TensorShape &in_tensor_shape, in LayoutTransferValidLayoutChangeCheck()
317 …idLayoutChangeCheck(const DeviceArrangement &in_device_arrangement, const TensorMap &in_tensor_map, in ValidLayoutChangeCheck()
|
| D | tensor_layout_test.cc | 218 TensorMap in_tensor_map = {1, -1}; in TEST_F() local
236 TensorMap in_tensor_map = {1, -1}; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | broadcast_to_info.cc | 68 TensorMap in_tensor_map; in InferTensorMap() local
|
| D | pack_info.cc | 121 TensorMap in_tensor_map; in InferTensorMap() local
|
| D | unique_info.cc | 46 TensorMap in_tensor_map; in InferTensorMap() local
|
| D | transpose_info.cc | 221 auto in_tensor_map = in_layout.tensor_map_before(); in CheckOutputLayout() local
|