Searched refs:dist_dataset (Results 1 – 6 of 6) sorted by relevance
70 dist_dataset = distribution.experimental_distribute_dataset(dataset)72 iterator = iter(dist_dataset)103 dist_dataset = distribution.experimental_distribute_dataset(dataset)105 iterator = iter(dist_dataset)148 dist_dataset = distribution.experimental_distribute_dataset(dataset)151 iterator = iter(dist_dataset)217 dist_dataset = distribution.experimental_distribute_dataset(dataset)219 @def_function.function(input_signature=[dist_dataset.element_spec])223 for x in dist_dataset:362 dist_dataset = distribution.experimental_distribute_dataset(dataset)[all …]
132 dist_dataset = distribution.experimental_distribute_dataset(dataset)134 for x in dist_dataset:146 dist_dataset = distribution.experimental_distribute_dataset(dataset)147 iterator = iter(dist_dataset)212 dist_dataset = distribution.experimental_distribute_dataset(dataset)214 for x in dist_dataset:236 dist_dataset = distribution.experimental_distribute_dataset(dataset)238 for x in dist_dataset:291 dist_dataset = distribution.experimental_distribute_dataset(dataset)293 for x in dist_dataset:[all …]
86 dist_dataset = distribution.experimental_distribute_dataset(dataset)88 for x in dist_dataset:114 dist_dataset = distribution.experimental_distribute_dataset(dataset)116 for x in dist_dataset:
308 dist_dataset = input_lib.get_distributed_dataset(dataset_or_input_fn,312 dist_dataset = input_lib.get_distributed_datasets_from_function(317 iterator = iter(dist_dataset)324 iterator = iter(dist_dataset)344 dist_dataset = input_lib.get_distributed_dataset(347 iterator = dataset_ops.make_one_shot_iterator(dist_dataset)559 dist_dataset = input_lib.get_distributed_dataset(dataset, input_workers,562 iterator = iter(dist_dataset)832 dist_dataset = distribution.experimental_distribute_dataset(dataset)835 distribution.experimental_local_results(x) for x in dist_dataset)[all …]
521 dist_dataset = default_strategy.experimental_distribute_dataset(523 next_val = next(iter(dist_dataset))526 dist_dataset = default_strategy.experimental_distribute_dataset(528 iterator = dist_dataset.make_initializable_iterator()
1451 dist_dataset = DistributedDatasetV1(1458 dist_dataset._cloned_datasets, input_workers, True) # pylint: disable=protected-access1461 … dist_dataset._enable_get_next_as_optional) # pylint: disable=protected-access1462 self._element_spec = dist_dataset.element_spec