Searched refs:_ReincarnatedPerDeviceGenerator (Results 1 – 1 of 1) sorted by relevance
147 class _ReincarnatedPerDeviceGenerator(dataset_ops.DatasetV2): class179 super(_ReincarnatedPerDeviceGenerator, self).__init__(variant_tensor)283 ds = _ReincarnatedPerDeviceGenerator(ds, self._incarnation_id)