Searched refs:create_sharding (Results 1 – 1 of 1) sorted by relevance
183 auto create_sharding = [](const Array<int64_t> tiles, in Subgroup() local235 return create_sharding(new_tiles, merged_types, metadata); in Subgroup()237 return create_sharding(tile_assignment, subgroup_types, metadata); in Subgroup()