Home
last modified time | relevance | path

Searched refs:_RebatchDataset (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Drebatch_dataset_test.py169 rebatched_dataset = distribute._RebatchDataset(
179 rebatched_dataset = distribute._RebatchDataset(
188 rebatched_dataset = distribute._RebatchDataset(
197 rebatched_dataset = distribute._RebatchDataset(
206 rebatched_dataset = distribute._RebatchDataset(
215 rebatched_dataset = distribute._RebatchDataset(
228 rebatched_dataset = distribute._RebatchDataset(
241 rebatched_dataset = distribute._RebatchDataset(
253 rebatched_dataset = distribute._RebatchDataset(
267 rebatched_dataset = distribute._RebatchDataset(
[all …]
Dauto_shard_dataset_test.py622 dataset = distribute._RebatchDataset(dataset, batch_sizes=[2, 1, 2])
648 worker_a_dataset = distribute._RebatchDataset(
657 worker_b_dataset = distribute._RebatchDataset(
666 worker_c_dataset = distribute._RebatchDataset(
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/
Drebatch_dataset_serialization_test.py54 return distribute._RebatchDataset(
/external/tensorflow/tensorflow/python/data/experimental/ops/
Ddistribute.py93 class _RebatchDataset(dataset_ops.UnaryDataset): class
152 super(_RebatchDataset, self).__init__(input_dataset, variant_tensor)
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib.py1067 return distribute._RebatchDataset(