Searched refs:_MapAndBatchDataset (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/data/experimental/ops/ |
| D | batching.py | 565 class _MapAndBatchDataset(dataset_ops.UnaryDataset): class
602 super(_MapAndBatchDataset, self).__init__(input_dataset, variant_tensor)
661 return _MapAndBatchDataset(dataset, map_func, batch_size,
725 return _MapAndBatchDataset(dataset, map_func, batch_size,
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | input_lib.py | 501 if isinstance(d, (dataset_ops.BatchDataset, batching._MapAndBatchDataset)):
517 (dataset_ops.BatchDataset, batching._MapAndBatchDataset))
521 elif isinstance(d, batching._MapAndBatchDataset):
|