Searched refs:new_input_tile_assignment (Results 1 – 1 of 1) sorted by relevance
2688 Array<int64_t> new_input_tile_assignment = sharding.tile_assignment(); in HandleReshape() local2689 new_input_tile_assignment.Reshape( in HandleReshape()2693 ? HloSharding::PartialTile(new_input_tile_assignment) in HandleReshape()2694 : HloSharding::Tile(new_input_tile_assignment); in HandleReshape()