Searched refs:_verify_dataset_shape (Results 1 – 1 of 1) sorted by relevance
735 self._verify_dataset_shape(dataset)786 def _verify_dataset_shape(self, dataset): member in TPUDatasetInfeedManager