Searched defs:sharding_dims (Results 1 – 2 of 2) sorted by relevance
22 def _chunk_sharding_specs_list_for_test(sharding_dims, seed=0): argument
1431 absl::Span<const int64_t> sharding_dims, in AlignShardingOnDims()