Searched defs:new_tile_dims (Results 1 – 2 of 2) sorted by relevance
55 auto new_tile_dims = group_tile_assignment.dimensions(); in PartialTile() local76 auto new_tile_dims = tile_assignment_last_dim_replicate.dimensions(); in PartialTile() local
89 std::vector<int64> new_tile_dims; in MergeSharding() local