Searched refs:per_device_dataset (Results 1 – 1 of 1) sorted by relevance
172 def __init__(self, per_device_dataset, incarnation_id): argument174 self._element_spec = per_device_dataset.element_spec175 self._init_func = per_device_dataset._init_func178 self._next_func = per_device_dataset._next_func179 self._next_captured_args = per_device_dataset._next_captured_args183 per_device_dataset._incarnation_id_index] = incarnation_id185 self._finalize_func = per_device_dataset._finalize_func186 self._finalize_captured_args = per_device_dataset._finalize_captured_args