| 25 | } |
| 26 | |
| 27 | OpSharding AssignDevice(int device) { |
| 28 | OpSharding result; |
| 29 | result.set_type(OpSharding::MAXIMAL); |
| 30 | result.add_tile_assignment_dimensions(1); |
| 31 | result.add_tile_assignment_devices(device); |
| 32 | return result; |
| 33 | } |
| 34 | |
| 35 | OpSharding Tile(const Shape& tile_shape, |
| 36 | const TileAssignment& tile_assignment) { |
no outgoing calls