Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/distribute/
Dcustom_training_loop_input_test.py134 dist_dataset = distribution.experimental_distribute_dataset(dataset)
148 dist_dataset = distribution.experimental_distribute_dataset(dataset)
171 distribution.experimental_distribute_dataset(dataset))
196 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
214 dist_dataset = distribution.experimental_distribute_dataset(dataset)
238 dist_dataset = distribution.experimental_distribute_dataset(dataset)
255 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
293 dist_dataset = distribution.experimental_distribute_dataset(dataset)
328 dist_dataset = distribution.experimental_distribute_dataset(dataset)
369 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)
352 dist_dataset = distribution.experimental_distribute_dataset(
385 return distribution.experimental_distribute_dataset(dataset)
435 dist_dataset = distribution.experimental_distribute_dataset(dataset)
[all …]
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.py344 dataset = strategy.experimental_distribute_dataset(dataset)
653 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
675 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
1026 dataset_item = next(iter(strategy.experimental_distribute_dataset(dataset)))
1039 dataset_item = next(iter(strategy.experimental_distribute_dataset(
1054 dataset_item = next(iter(strategy.experimental_distribute_dataset(
1071 iter(strategy.experimental_distribute_dataset(dataset))
1084 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(
Dinput_lib_test.py558 dist_dataset = distribution.experimental_distribute_dataset(dataset)
855 dist_dataset = distribution.experimental_distribute_dataset(dataset)
884 dist_dataset = distribution.experimental_distribute_dataset(dataset)
916 dist_dataset = distribution.experimental_distribute_dataset(dataset)
1022 dist_dataset = strategy.experimental_distribute_dataset(dataset)
1203 ds = distribution.experimental_distribute_dataset(
1252 ds = distribution.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))
159 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
187 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
234 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
270 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
337 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
371 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
399 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
[all …]
Dctl_correctness_test.py157 dataset = strategy.experimental_distribute_dataset(dataset)
207 dataset = strategy.experimental_distribute_dataset(dataset)
Dcustom_training_loop_metrics_test.py70 dataset = distribution.experimental_distribute_dataset(dataset)
/external/tensorflow/tensorflow/python/tpu/
Dtpu_embedding_v2_test.py464 strategy.experimental_distribute_dataset(
576 strategy.experimental_distribute_dataset(
581 strategy.experimental_distribute_dataset(
606 strategy.experimental_distribute_dataset(
611 strategy.experimental_distribute_dataset(
636 strategy.experimental_distribute_dataset(
641 strategy.experimental_distribute_dataset(
666 strategy.experimental_distribute_dataset(
690 strategy.experimental_distribute_dataset(
715 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.experimental.-multi-worker-mirrored-strategy.pbtxt37 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-one-device-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-strategy.pbtxt35 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt36 name: "experimental_distribute_dataset"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.distribute.experimental.-central-storage-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-strategy.pbtxt35 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-t-p-u-strategy.pbtxt40 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-parameter-server-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-one-device-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"
Dtensorflow.distribute.-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"

12