Searched refs:new_tile_dimension (Results 1 – 1 of 1) sorted by relevance
511 const int64 new_tile_dimension = in GatherOutputSharding() local514 index_sharding.tile_assignment().dim(new_tile_dimension)); in GatherOutputSharding()