Searched refs:_PerDeviceGenerator (Results 1 – 1 of 1) sorted by relevance
38 class _PerDeviceGenerator(dataset_ops.DatasetV2): class140 super(_PerDeviceGenerator, self).__init__(variant_tensor)274 ds = _PerDeviceGenerator(i, self._multi_device_iterator_resource,549 ds = _PerDeviceGenerator(i, self._multi_device_iterator_resource,