Searched refs:new_input_tile_assignment (Results 1 – 1 of 1) sorted by relevance
2031 Array<int64> new_input_tile_assignment = sharding.tile_assignment(); in HandleReshape() local2032 new_input_tile_assignment.Reshape( in HandleReshape()2036 ? HloSharding::PartialTile(new_input_tile_assignment) in HandleReshape()2037 : HloSharding::Tile(new_input_tile_assignment); in HandleReshape()