Searched defs:tensor_pow_size (Results 1 – 3 of 3) sorted by relevance
260 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() local313 int64_t tensor_pow_size = tensor_pow_size_min; in TEST_F() local
245 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
152 int64_t device_pow_size, int64_t tensor_pow_size, int64_t max_device_dim, int64_t max_shape_dim, in GenerateValidLayoutByDeviceSizeAndTensorSize()