Home
last modified time | relevance | path

Searched refs:input_lib (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/distribute/
Done_device_strategy.py24 from tensorflow.python.distribute import input_lib
265 return input_lib.InputWorkers([(self._input_device, (self._device,))])
267 return input_lib.InputWorkers([(self._input_device,
293 return input_lib.DatasetIterator(dataset, self._input_workers,
300 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
322 return input_lib.get_distributed_dataset(
335 return input_lib.get_distributed_datasets_from_function(
354 ctx = input_lib.MultiStepContext()
Dinput_lib_test.py36 from tensorflow.python.distribute import input_lib
91 iterator = input_lib.InputFunctionIterator(
97 iterator = input_lib.DatasetIterator(
114 return input_lib.DistributedDataset(
121 return input_lib.DistributedDatasetV1(
183 input_workers = input_lib.InputWorkers(worker_device_pairs)
207 if isinstance(dataset, input_lib.DistributedDatasetV1):
301 input_workers = input_lib.InputWorkers(worker_device_pairs)
306 input_workers = input_lib.InputWorkers(worker_device_pairs)
308 dist_dataset = input_lib.get_distributed_dataset(dataset_or_input_fn,
[all …]
Dmirrored_strategy.py29 from tensorflow.python.distribute import input_lib
443 return input_lib.InputWorkers(self._input_workers_devices)
454 return input_lib.InputWorkers(self._input_workers_devices)
457 return input_lib.InputWorkers([
462 return input_lib.InputWorkers(self._input_workers_devices)
538 return input_lib.DatasetIterator(
555 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
567 return input_lib.get_distributed_dataset(
587 return input_lib.get_distributed_datasets_from_function(
606 ctx = input_lib.MultiStepContext()
Dparameter_server_strategy.py28 from tensorflow.python.distribute import input_lib
327 return input_lib.InputWorkers(
330 return input_lib.InputWorkers(
342 return input_lib.get_distributed_dataset(
349 return input_lib.DatasetIterator(
372 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
395 return input_lib.get_distributed_datasets_from_function(
Done_device_strategy_test.py24 from tensorflow.python.distribute import input_lib
132 if isinstance(dataset, input_lib.DistributedDatasetV1):
153 if isinstance(dataset, input_lib.DistributedDatasetV1):
Dcollective_all_reduce_strategy.py35 from tensorflow.python.distribute import input_lib
526 return input_lib.InputWorkers([(host_device, self.worker_devices)])
528 return input_lib.InputWorkers([(
602 return input_lib.get_distributed_dataset(
618 return input_lib.get_distributed_datasets_from_function(
638 return input_lib.DatasetIterator(
651 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
Dtpu_strategy.py38 from tensorflow.python.distribute import input_lib
879 input_workers = input_lib.InputWorkers(
881 return input_lib.DatasetIterator(
892 input_workers = input_lib.InputWorkers(
900 return input_lib.InputFunctionIterator(
913 return input_lib.InputWorkers(
916 return input_lib.InputWorkers(
945 return input_lib.get_distributed_dataset(
968 distributed_dataset = input_lib.get_distributed_datasets_from_function(
996 ctx = input_lib.MultiStepContext()
DBUILD167 ":input_lib",
320 ":input_lib",
349 ":input_lib",
389 ":input_lib",
414 ":input_lib",
507 name = "input_lib",
508 srcs = ["input_lib.py"],
586 ":input_lib",
1013 ":input_lib",
1045 ":input_lib",
[all …]
Ddistribute_lib_test.py28 from tensorflow.python.distribute import input_lib
72 self._input_workers = input_lib.InputWorkers(worker_device_pairs)
87 return input_lib.InputFunctionIterator(input_fn, self._input_workers,
104 ctx = input_lib.MultiStepContext()
Dcollective_all_reduce_strategy_test.py35 from tensorflow.python.distribute import input_lib
300 if isinstance(dataset, input_lib.DistributedDatasetV1):
321 if isinstance(dataset, input_lib.DistributedDatasetV1):
Dparameter_server_strategy_test.py33 from tensorflow.python.distribute import input_lib
790 if isinstance(dataset, input_lib.DistributedDatasetV1):
812 if isinstance(dataset, input_lib.DistributedDatasetV1):
Dmirrored_strategy_test.py36 from tensorflow.python.distribute import input_lib
247 if isinstance(dataset, input_lib.DistributedDatasetV1):
269 if isinstance(dataset, input_lib.DistributedDatasetV1):
/external/tensorflow/tensorflow/python/autograph/operators/
DBUILD89 "//tensorflow/python/distribute:input_lib",
125 "//tensorflow/python/distribute:input_lib",
Dpy_builtins.py54 input_lib = lazy_loader.LazyLoader( variable
397 s, (input_lib.DistributedIterator, input_lib.DistributedDataset)):
Dcontrol_flow_deprecated_py2.py93 input_lib = lazy_loader.LazyLoader( variable
353 if isinstance(iter_, input_lib.DistributedIterator):
358 if isinstance(iter_, input_lib.DistributedDataset):
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.-distributed-dataset.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.input_lib.DistributedDatasetInterface\'>"
Dtensorflow.distribute.-distributed-iterator.pbtxt3 is_instance: "<class \'tensorflow.python.distribute.input_lib.DistributedIteratorInterface\'>"
/external/tensorflow/tensorflow/python/training/
Dsupervisor_test.py49 from tensorflow.python.training import input as input_lib unknown
275 filename_queue = input_lib.string_input_producer(
294 filename_queue = input_lib.string_input_producer(
299 shuff_rec = input_lib.shuffle_batch(rec, 1, 6, 4)
314 filename_queue = input_lib.string_input_producer(
319 shuff_rec = input_lib.shuffle_batch(rec, 1, 6, 4)
/external/tensorflow/tensorflow/python/distribute/coordinator/
DBUILD23 "//tensorflow/python/distribute:input_lib",
Dcluster_coordinator.py36 from tensorflow.python.distribute import input_lib
1217 input_workers = input_lib.InputWorkers([
/external/tensorflow/tensorflow/python/autograph/converters/
DBUILD121 "//tensorflow/python/distribute:input_lib",
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining_distributed_v1.py26 from tensorflow.python.distribute import input_lib
454 batch_size, _, prefetch_buffer = input_lib._get_dataset_attributes(dataset)
Ddata_adapter.py36 from tensorflow.python.distribute import input_lib
1661 return isinstance(ds, input_lib.DistributedDatasetInterface)
DBUILD57 "//tensorflow/python/distribute:input_lib",
/external/tensorflow/tensorflow/python/kernel_tests/
Dreader_ops_test.py39 from tensorflow.python.training import input as input_lib unknown
788 filename_queue = input_lib.string_input_producer(
825 filename_queue = input_lib.string_input_producer(

12