/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"
|