Searched refs:_check_spec (Results 1 – 1 of 1) sorted by relevance
919 def _check_spec(self, element_spec): member in TPUExtended943 self._check_spec(dataset.element_spec)976 self._check_spec(distributed_dataset.element_spec)