Searched refs:_wrap_dataset (Results 1 – 1 of 1) sorted by relevance
105 def _wrap_dataset(self, member in DistributedIteratorTestBase196 dataset = self._wrap_dataset(1061 dataset = self._wrap_dataset(input_type, dataset_or_input_fn,