Home
last modified time | relevance | path

Searched refs:per_device_dataset (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/data/ops/
Dmulti_device_iterator_ops.py154 def __init__(self, per_device_dataset, incarnation_id): argument
156 self._structure = per_device_dataset._structure
158 self._init_func = per_device_dataset._init_func
161 self._next_func = per_device_dataset._next_func
162 self._next_captured_args = per_device_dataset._next_captured_args
166 per_device_dataset._incarnation_id_index] = incarnation_id
168 self._finalize_func = per_device_dataset._finalize_func
169 self._finalize_captured_args = per_device_dataset._finalize_captured_args