Creates a dataset on a given `device` given a graph def.
| 110 | |
| 111 | |
| 112 | class _RemoteDataset(dataset_ops.DatasetSource): |
| 113 | """Creates a dataset on a given `device` given a graph def.""" |
| 114 | |
| 115 | def __init__(self, graph_def, device, element_spec): |
| 116 | self._elem_spec = element_spec |
| 117 | with ops.device(device): |
| 118 | variant_tensor = ged_ops.dataset_from_graph(graph_def) |
| 119 | super(_RemoteDataset, self).__init__(variant_tensor) |
| 120 | |
| 121 | @property |
| 122 | def element_spec(self): |
| 123 | return self._elem_spec |
| 124 | |
| 125 | |
| 126 | def replicate(dataset, devices): |