Home
last modified time | relevance | path

Searched defs:tensor_pow_size (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/
Dreshape_layout_transfer_test.cc260 void ValidInferUnifiedLayoutCheckAll(int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_… in ValidInferUnifiedLayoutCheckAll()
295 int64_t tensor_pow_size = tensor_pow_size_min; in TEST_F() local
313 int64_t tensor_pow_size = tensor_pow_size_min; in TEST_F() local
Dredistribution_layout_transfer_test.cc245 void ValidRedistributionLayoutCheckAll(int64_t device_pow_size, int64_t tensor_pow_size, int64_t ma… in ValidRedistributionLayoutCheckAll()
281 int64_t tensor_pow_size = tensor_pow_size_min; in TEST_F() local
Dutil_layout_gen_test.cc152 int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_device_dim, int64_t max_shape_dim, in GenerateValidLayoutByDeviceSizeAndTensorSize()