Searched defs:sharding_dim (Results 1 – 2 of 2) sorted by relevance
17 def generate_chunk_sharding_specs_for_test(sharding_dim): argument
369 def _infer_chunk_sharding_spec_case(self, placements, sharding_dim, st_size): argument