Searched refs:_MapAndBatchDataset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 200 return _MapAndBatchDataset(dataset, map_func, batch_size,
261 return _MapAndBatchDataset(dataset, map_func, batch_size,
327 class _MapAndBatchDataset(dataset_ops.UnaryDataset): class
370 super(_MapAndBatchDataset, self).__init__(input_dataset, variant_tensor)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 1935 if isinstance(d, (dataset_ops.BatchDataset, batching._MapAndBatchDataset)):
1951 (dataset_ops.BatchDataset, batching._MapAndBatchDataset))
1955 elif isinstance(d, batching._MapAndBatchDataset):
|