Home
last modified time | relevance | path

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

123

/aosp_15_r20/external/tensorflow/tensorflow/python/distribute/
H A Dcustom_training_loop_input_test.py130 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 Dinput_lib_type_spec_test.py66 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 Dtpu_strategy_test.py428 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 Dcentral_storage_strategy.py73 …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 Dcustom_training_loop_gradient_test.py82 dist_dataset = distribution.experimental_distribute_dataset(dataset)
110 dist_dataset = distribution.experimental_distribute_dataset(dataset)
H A Dinput_lib_test.py515 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 Done_device_strategy_test.py124 dataset = distribution.experimental_distribute_dataset(
145 dataset = distribution.experimental_distribute_dataset(
H A Done_device_strategy.py81 …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 Dparameter_server_strategy.py118 def experimental_distribute_dataset(self, dataset, options=None): member in ParameterServerStrategyV1
128 self).experimental_distribute_dataset(dataset=dataset,
H A Dparameter_server_strategy_test.py755 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 Dtpu_embedding_v2_invalid_input_test.py81 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 Dtpu_embedding_v2_hd_valid_input_test.py38 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 Dtpu_embedding_v2_valid_input_test.py48 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 Dtpu_embedding_v2_enqueue_mode_test.py45 strategy.experimental_distribute_dataset(
84 strategy.experimental_distribute_dataset(
117 strategy.experimental_distribute_dataset(
145 strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v2_sequence_feature_test.py40 strategy.experimental_distribute_dataset(
71 strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v1_correctness_test.py131 dist = strategy.experimental_distribute_dataset(
279 strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v2_hd_invalid_input_test.py42 strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v2_correctness_dense_lookup_test.py37 dist = strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v2_correctness_sequence_feature_test.py62 strategy.experimental_distribute_dataset(
H A Dtpu_embedding_v2_correctness_base_test.py58 dist = strategy.experimental_distribute_dataset(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_factory_ops_test.py139 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 Dcluster_coordinator_test.py654 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 Dfactory_ops_test.py61 dist_dataset = strategy.experimental_distribute_dataset(sparse_ds)
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.distribute.-strategy.pbtxt35 name: "experimental_distribute_dataset"
H A Dtensorflow.distribute.experimental.-multi-worker-mirrored-strategy.pbtxt36 name: "experimental_distribute_dataset"

123