Searched refs:_ThreadPoolDataset (Results 1 – 1 of 1) sorted by relevance
63 class _ThreadPoolDataset(dataset_ops.UnaryUnchangedStructureDataset): class73 super(_ThreadPoolDataset, self).__init__(input_dataset, variant_tensor)90 return _ThreadPoolDataset(dataset, thread_pool)