Searched refs:_wrap_dataset (Results 1 – 1 of 1) sorted by relevance
103 def _wrap_dataset(self, member in DistributedIteratorTestBase194 dataset = self._wrap_dataset(1125 dataset = self._wrap_dataset(input_type, dataset_or_input_fn,