Searched refs:_MapOnGpuDataset (Results 1 – 1 of 1) sorted by relevance
228 class _MapOnGpuDataset(dataset_ops.UnaryDataset): class247 super(_MapOnGpuDataset, self).__init__(input_dataset, variant_tensor)279 return _MapOnGpuDataset(dataset, map_func)