Home
last modified time | relevance | path

Searched refs:experimental_distribute_dataset (Results 1 – 25 of 51) sorted by relevance

123

/external/tensorflow/tensorflow/python/distribute/
Dcustom_training_loop_input_test.py132 dist_dataset = distribution.experimental_distribute_dataset(dataset)
146 dist_dataset = distribution.experimental_distribute_dataset(dataset)
169 distribution.experimental_distribute_dataset(dataset))
194 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
212 dist_dataset = distribution.experimental_distribute_dataset(dataset)
236 dist_dataset = distribution.experimental_distribute_dataset(dataset)
253 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
291 dist_dataset = distribution.experimental_distribute_dataset(dataset)
326 dist_dataset = distribution.experimental_distribute_dataset(dataset)
367 dist_dataset = distribution.experimental_distribute_dataset(dataset)
[all …]
Dinput_lib_type_spec_test.py70 dist_dataset = distribution.experimental_distribute_dataset(dataset)
103 dist_dataset = distribution.experimental_distribute_dataset(dataset)
148 dist_dataset = distribution.experimental_distribute_dataset(dataset)
186 ds = distribution.experimental_distribute_dataset(ds)
217 dist_dataset = distribution.experimental_distribute_dataset(dataset)
246 dist_ds1 = distribution.experimental_distribute_dataset(ds1)
247 dist_ds2 = distribution.experimental_distribute_dataset(ds2)
362 dist_dataset = distribution.experimental_distribute_dataset(dataset)
432 dist_dataset = distribution.experimental_distribute_dataset(
494 dist_dataset = distribution.experimental_distribute_dataset(
Dcentral_storage_strategy.py77 …def experimental_distribute_dataset(self, dataset, options=None): # pylint: disable=useless-super… member in CentralStorageStrategy
112 return super(CentralStorageStrategy, self).experimental_distribute_dataset(
Dcustom_training_loop_gradient_test.py86 dist_dataset = distribution.experimental_distribute_dataset(dataset)
114 dist_dataset = distribution.experimental_distribute_dataset(dataset)
Done_device_strategy_test.py127 dataset = distribution.experimental_distribute_dataset(
148 dataset = distribution.experimental_distribute_dataset(
Dtpu_strategy_test.py291 dataset = strategy.experimental_distribute_dataset(dataset)
600 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
622 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
877 dataset_item = next(iter(strategy.experimental_distribute_dataset(dataset)))
890 dataset_item = next(iter(strategy.experimental_distribute_dataset(
905 dataset_item = next(iter(strategy.experimental_distribute_dataset(
922 iter(strategy.experimental_distribute_dataset(dataset))
935 iter(strategy.experimental_distribute_dataset(dataset))
Done_device_strategy.py83 …def experimental_distribute_dataset(self, dataset, options=None): # pylint: disable=useless-super… member in OneDeviceStrategy
109 return super(OneDeviceStrategy, self).experimental_distribute_dataset(
/external/tensorflow/tensorflow/python/keras/distribute/
Dcustom_training_loop_models_test.py70 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
94 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
132 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
157 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
185 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
232 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
268 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
335 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
369 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
397 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
[all …]
Dctl_correctness_test.py154 dataset = strategy.experimental_distribute_dataset(dataset)
197 dataset = strategy.experimental_distribute_dataset(dataset)
Dcustom_training_loop_metrics_test.py73 dataset = distribution.experimental_distribute_dataset(dataset)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_test.py461 data = next(iter(strategy.experimental_distribute_dataset(
573 sparse_iter = iter(strategy.experimental_distribute_dataset(
577 ragged_iter = iter(strategy.experimental_distribute_dataset(
601 sparse_iter = iter(strategy.experimental_distribute_dataset(
605 ragged_iter = iter(strategy.experimental_distribute_dataset(
629 sparse_iter = iter(strategy.experimental_distribute_dataset(
633 ragged_iter = iter(strategy.experimental_distribute_dataset(
657 sparse_iter = iter(strategy.experimental_distribute_dataset(
680 sparse_iter = iter(strategy.experimental_distribute_dataset(
704 sparse_iter = iter(strategy.experimental_distribute_dataset(
[all …]
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt37 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-strategy.pbtxt35 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-multi-worker-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt36 name: "experimental_distribute_dataset"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.-strategy.pbtxt35 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt40 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-one-device-strategy.pbtxt36 name: "experimental_distribute_dataset"

123