Searched refs:per_device_dataset (Results 1 – 1 of 1) sorted by relevance
158 def __init__(self, per_device_dataset, incarnation_id): argument160 self._element_spec = per_device_dataset.element_spec161 self._init_func = per_device_dataset._init_func164 self._next_func = per_device_dataset._next_func165 self._next_captured_args = per_device_dataset._next_captured_args169 per_device_dataset._incarnation_id_index] = incarnation_id171 self._finalize_func = per_device_dataset._finalize_func172 self._finalize_captured_args = per_device_dataset._finalize_captured_args