Searched defs:max_shape_dim (Results 1 – 3 of 3) sorted by relevance
261 int64_t max_shape_dim) { in ValidInferUnifiedLayoutCheckAll()292 const int64_t max_shape_dim = 5; in TEST_F() local310 const int64_t max_shape_dim = 5; in TEST_F() local
246 int64_t max_shape_dim) { in ValidRedistributionLayoutCheckAll()278 const int64_t max_shape_dim = 5; 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()