Searched refs:_create_high_dimensional_ragged_dataset (Results 1 – 2 of 2) sorted by relevance
50 ragged = self._create_high_dimensional_ragged_dataset(strategy)
264 def _create_high_dimensional_ragged_dataset(self, member in TPUEmbeddingBaseTest