Searched refs:_RebatchDataset (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | rebatch_dataset_test.py | 45 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
59 batching._RebatchDataset(dataset, num_workers=4)
66 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=5)
74 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
84 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
93 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
107 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
133 rebatched_dataset = batching._RebatchDataset(dataset, 4)
147 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
161 rebatched_dataset = batching._RebatchDataset(dataset, num_workers=4)
[all …]
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/serialization/ |
| D | rebatch_dataset_serialization_test.py | 32 return batching._RebatchDataset(
|
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 731 class _RebatchDataset(dataset_ops.UnaryDataset): class
762 super(_RebatchDataset, self).__init__(input_dataset, variant_tensor)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 310 … dataset = batching._RebatchDataset(dataset, split_batch_by) # pylint: disable=protected-access
|