Searched defs:max_device_dim (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | reshape_layout_transfer_test.cc | 260 …nferUnifiedLayoutCheckAll(int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_device_dim, in ValidInferUnifiedLayoutCheckAll() 291 const int64_t max_device_dim = 5; in TEST_F() local 309 const int64_t max_device_dim = 5; in TEST_F() local
|
| D | redistribution_layout_transfer_test.cc | 245 …istributionLayoutCheckAll(int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_device_dim, in ValidRedistributionLayoutCheckAll() 277 const int64_t max_device_dim = 5; in TEST_F() local
|
| D | util_layout_gen_test.cc | 152 int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_device_dim, int64_t max_shape_dim, in GenerateValidLayoutByDeviceSizeAndTensorSize()
|