| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ |
| H A D | custom_training_loop_input_test.py | 130 dist_dataset = distribution.experimental_distribute_dataset(dataset)
144 dist_dataset = distribution.experimental_distribute_dataset(dataset)
167 distribution.experimental_distribute_dataset(dataset))
192 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
210 dist_dataset = distribution.experimental_distribute_dataset(dataset)
234 dist_dataset = distribution.experimental_distribute_dataset(dataset)
251 input_iterator = iter(distribution.experimental_distribute_dataset(dataset))
289 dist_dataset = distribution.experimental_distribute_dataset(dataset)
324 dist_dataset = distribution.experimental_distribute_dataset(dataset)
365 dist_dataset = distribution.experimental_distribute_dataset(dataset)
[all …]
|
| H A D | input_lib_type_spec_test.py | 66 dist_dataset = distribution.experimental_distribute_dataset(dataset)
99 dist_dataset = distribution.experimental_distribute_dataset(dataset)
144 dist_dataset = distribution.experimental_distribute_dataset(dataset)
182 ds = distribution.experimental_distribute_dataset(ds)
213 dist_dataset = distribution.experimental_distribute_dataset(dataset)
242 dist_ds1 = distribution.experimental_distribute_dataset(ds1)
243 dist_ds2 = distribution.experimental_distribute_dataset(ds2)
348 dist_dataset = distribution.experimental_distribute_dataset(
381 return distribution.experimental_distribute_dataset(dataset)
431 dist_dataset = distribution.experimental_distribute_dataset(dataset)
[all …]
|
| H A D | tpu_strategy_test.py | 428 dataset = strategy.experimental_distribute_dataset(dataset)
737 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
759 input_iterator = iter(strategy.experimental_distribute_dataset(dataset))
1111 dist_dataset = strategy.experimental_distribute_dataset(
1132 dist_dataset = strategy.experimental_distribute_dataset(
1159 dataset_item = next(iter(strategy.experimental_distribute_dataset(dataset)))
1172 dataset_item = next(iter(strategy.experimental_distribute_dataset(
1187 dataset_item = next(iter(strategy.experimental_distribute_dataset(
1204 iter(strategy.experimental_distribute_dataset(dataset))
1217 iter(strategy.experimental_distribute_dataset(dataset))
|
| H A D | central_storage_strategy.py | 73 …def experimental_distribute_dataset(self, dataset, options=None): # pylint: disable=useless-super… member in CentralStorageStrategy
108 return super(CentralStorageStrategy, self).experimental_distribute_dataset(
|
| H A D | custom_training_loop_gradient_test.py | 82 dist_dataset = distribution.experimental_distribute_dataset(dataset)
110 dist_dataset = distribution.experimental_distribute_dataset(dataset)
|
| H A D | input_lib_test.py | 515 dist_dataset = distribution.experimental_distribute_dataset(dataset)
812 dist_dataset = distribution.experimental_distribute_dataset(dataset)
841 dist_dataset = distribution.experimental_distribute_dataset(dataset)
873 dist_dataset = distribution.experimental_distribute_dataset(dataset)
911 dist_dataset = distribution.experimental_distribute_dataset(dataset)
951 dist_dataset = distribution.experimental_distribute_dataset(dataset)
1059 dist_dataset = strategy.experimental_distribute_dataset(dataset)
1243 ds = distribution.experimental_distribute_dataset(
1287 ds = distribution.experimental_distribute_dataset(
|
| H A D | one_device_strategy_test.py | 124 dataset = distribution.experimental_distribute_dataset(
145 dataset = distribution.experimental_distribute_dataset(
|
| H A D | one_device_strategy.py | 81 …def experimental_distribute_dataset(self, dataset, options=None): # pylint: disable=useless-super… member in OneDeviceStrategy
107 return super(OneDeviceStrategy, self).experimental_distribute_dataset(
|
| H A D | parameter_server_strategy.py | 118 def experimental_distribute_dataset(self, dataset, options=None): member in ParameterServerStrategyV1
128 self).experimental_distribute_dataset(dataset=dataset,
|
| H A D | parameter_server_strategy_test.py | 755 strategy.experimental_distribute_dataset,
784 …dataset = distribution.experimental_distribute_dataset( # pylint: disable=assignment-from-no-retu…
807 …dataset = distribution.experimental_distribute_dataset( # pylint: disable=assignment-from-no-retu…
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/tpu/tests/ |
| H A D | tpu_embedding_v2_invalid_input_test.py | 81 strategy.experimental_distribute_dataset(
105 strategy.experimental_distribute_dataset(
110 strategy.experimental_distribute_dataset(
149 strategy.experimental_distribute_dataset(
186 dense_iter = iter(strategy.experimental_distribute_dataset(dataset))
210 dense_iter = iter(strategy.experimental_distribute_dataset(dataset))
|
| H A D | tpu_embedding_v2_hd_valid_input_test.py | 38 strategy.experimental_distribute_dataset(
45 strategy.experimental_distribute_dataset(
52 strategy.experimental_distribute_dataset(
81 strategy.experimental_distribute_dataset(
92 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_valid_input_test.py | 48 strategy.experimental_distribute_dataset(
99 strategy.experimental_distribute_dataset(
104 strategy.experimental_distribute_dataset(
128 strategy.experimental_distribute_dataset(
185 strategy.experimental_distribute_dataset(
270 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_enqueue_mode_test.py | 45 strategy.experimental_distribute_dataset(
84 strategy.experimental_distribute_dataset(
117 strategy.experimental_distribute_dataset(
145 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_sequence_feature_test.py | 40 strategy.experimental_distribute_dataset(
71 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v1_correctness_test.py | 131 dist = strategy.experimental_distribute_dataset(
279 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_hd_invalid_input_test.py | 42 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_correctness_dense_lookup_test.py | 37 dist = strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_correctness_sequence_feature_test.py | 62 strategy.experimental_distribute_dataset(
|
| H A D | tpu_embedding_v2_correctness_base_test.py | 58 dist = strategy.experimental_distribute_dataset(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_factory_ops_test.py | 139 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)
177 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)
222 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)
283 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)
338 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/coordinator/ |
| H A D | cluster_coordinator_test.py | 654 return self.strategy.experimental_distribute_dataset(dataset)
1313 return self.strategy.experimental_distribute_dataset(dataset)
1339 distributed_dataset = self.strategy.experimental_distribute_dataset(
1389 return self.strategy.experimental_distribute_dataset(dataset)
1444 return self.strategy.experimental_distribute_dataset(dataset)
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | factory_ops_test.py | 61 dist_dataset = strategy.experimental_distribute_dataset(sparse_ds)
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.distribute.-strategy.pbtxt | 35 name: "experimental_distribute_dataset"
|
| H A D | tensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt | 36 name: "experimental_distribute_dataset"
|