Searched refs:input_lib_v1 (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_util.py | 19 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
69 return input_lib_v1.DistributedDatasetV1(
141 return input_lib_v1.DistributedDatasetsFromFunctionV1(
|
| D | one_device_strategy_test.py | 23 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
129 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
150 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
|
| D | one_device_strategy.py | 23 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
291 return input_lib_v1.DatasetIterator(dataset, self._input_workers,
298 return input_lib_v1.InputFunctionIterator(input_fn, self._input_workers,
|
| D | parameter_server_strategy.py | 33 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
348 return input_lib_v1.DatasetIterator(
371 return input_lib_v1.InputFunctionIterator(input_fn, self._input_workers,
|
| D | mirrored_strategy.py | 36 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
543 return input_lib_v1.DatasetIterator(
560 return input_lib_v1.InputFunctionIterator(input_fn, self._input_workers,
|
| D | collective_all_reduce_strategy.py | 41 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
697 return input_lib_v1.DatasetIterator(
710 return input_lib_v1.InputFunctionIterator(input_fn, self._input_workers,
|
| D | collective_all_reduce_strategy_test.py | 40 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
323 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
342 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
|
| D | parameter_server_strategy_test.py | 36 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
787 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
810 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
|
| D | input_lib_test.py | 38 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
91 iterator = input_lib_v1.InputFunctionIterator(dataset_or_input_fn,
95 iterator = input_lib_v1.DatasetIterator(
119 return input_lib_v1.DistributedDatasetV1(
205 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
|
| D | distribute_lib_test.py | 27 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
84 return input_lib_v1.InputFunctionIterator(input_fn, self._input_workers,
|
| D | tpu_strategy.py | 43 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
938 return input_lib_v1.DatasetIterator(
958 return input_lib_v1.InputFunctionIterator(input_fn, input_workers,
|
| D | mirrored_strategy_test.py | 39 from tensorflow.python.distribute.v1 import input_lib as input_lib_v1 unknown
271 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
293 if isinstance(dataset, input_lib_v1.DistributedDatasetV1):
|