Searched refs:_get_output_classes (Results 1 – 1 of 1) sorted by relevance
615 output_classes=self._get_output_classes(ds_fn))642 nest.flatten(self._get_output_classes(ds_fn))):657 for output_class in nest.flatten(self._get_output_classes(ds_fn)):677 def _get_output_classes(self, ds_fn): member in DatasetSerializationTestBase