Python representation of a xla.OpSharding protobuf.
| 1841 | |
| 1842 | |
| 1843 | class OpSharding(object): |
| 1844 | """Python representation of a xla.OpSharding protobuf.""" |
| 1845 | __slots__ = ('type', 'tile_assignment_dimensions', 'tile_assignment_devices', |
| 1846 | 'tuple_shardings') |
| 1847 | |
| 1848 | Type = _xla.OpSharding_Type |
| 1849 | |
| 1850 | def __init__(self): |
| 1851 | self.type = self.Type.REPLICATED |
| 1852 | self.tile_assignment_dimensions = [] |
| 1853 | self.tile_assignment_devices = [] |
| 1854 | self.tuple_shardings = [] |
| 1855 | |
| 1856 | |
| 1857 | class PrecisionConfig(object): |