Searched refs:dist_dataset (Results 1 – 10 of 10) sorted by relevance
66 dist_dataset = distribution.experimental_distribute_dataset(dataset)68 iterator = iter(dist_dataset)99 dist_dataset = distribution.experimental_distribute_dataset(dataset)101 iterator = iter(dist_dataset)144 dist_dataset = distribution.experimental_distribute_dataset(dataset)147 iterator = iter(dist_dataset)213 dist_dataset = distribution.experimental_distribute_dataset(dataset)215 @def_function.function(input_signature=[dist_dataset.element_spec])219 for x in dist_dataset:348 dist_dataset = distribution.experimental_distribute_dataset([all …]
130 dist_dataset = distribution.experimental_distribute_dataset(dataset)132 for x in dist_dataset:144 dist_dataset = distribution.experimental_distribute_dataset(dataset)145 iterator = iter(dist_dataset)210 dist_dataset = distribution.experimental_distribute_dataset(dataset)212 for x in dist_dataset:234 dist_dataset = distribution.experimental_distribute_dataset(dataset)236 for x in dist_dataset:289 dist_dataset = distribution.experimental_distribute_dataset(dataset)291 for x in dist_dataset:[all …]
299 dist_dataset = input_util.get_distributed_dataset(302 iterator = dataset_ops.make_one_shot_iterator(dist_dataset)499 dist_dataset = input_util.get_distributed_dataset(dataset, input_workers,502 iterator = iter(dist_dataset)515 dist_dataset = distribution.experimental_distribute_dataset(dataset)517 iterator = iter(dist_dataset)542 iterator = iter(dist_dataset)812 dist_dataset = distribution.experimental_distribute_dataset(dataset)815 distribution.experimental_local_results(x) for x in dist_dataset)817 distribution.experimental_local_results(x) for x in dist_dataset)[all …]
82 dist_dataset = distribution.experimental_distribute_dataset(dataset)84 for x in dist_dataset:110 dist_dataset = distribution.experimental_distribute_dataset(dataset)112 for x in dist_dataset:
538 dist_dataset = default_strategy.experimental_distribute_dataset(540 next_val = next(iter(dist_dataset))543 dist_dataset = default_strategy.experimental_distribute_dataset(545 iterator = dist_dataset.make_initializable_iterator()
1111 dist_dataset = strategy.experimental_distribute_dataset(1114 dist_iterator = iter(dist_dataset)1132 dist_dataset = strategy.experimental_distribute_dataset(1135 dist_iterator = iter(dist_dataset)
139 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)140 ds = iter(dist_dataset)177 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)184 for x in dist_dataset:222 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)223 ds = iter(dist_dataset)283 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)292 for x in dist_dataset:338 dist_dataset = strategy.experimental_distribute_dataset(ragged_ds)350 for x in dist_dataset:
257 dist_dataset = DistributedDatasetV1(265 dist_dataset._cloned_datasets, input_workers)268 dist_dataset.cardinality,269 dist_dataset._enable_get_next_as_optional)270 self._element_spec = dist_dataset.element_spec
61 dist_dataset = strategy.experimental_distribute_dataset(sparse_ds)62 ds = iter(dist_dataset)
1070 dist_dataset = strategy.experimental_distribute_dataset(ds)1072 per_replica_result = next(iter(dist_dataset))