Searched refs:_create_high_dimensional_sparse_dataset (Results 1 – 5 of 5) sorted by relevance
35 dataset = self._create_high_dimensional_sparse_dataset(strategy)40 dataset = self._create_high_dimensional_sparse_dataset(strategy)97 iter(self._create_high_dimensional_sparse_dataset(strategy)))
122 dataset = self._create_high_dimensional_sparse_dataset(strategy)127 dataset = self._create_high_dimensional_sparse_dataset(strategy)167 iter(self._create_high_dimensional_sparse_dataset(strategy)))
43 sparse = self._create_high_dimensional_sparse_dataset(strategy)79 sparse = self._create_high_dimensional_sparse_dataset(strategy)
40 sparse = self._create_high_dimensional_sparse_dataset(strategy)
230 def _create_high_dimensional_sparse_dataset(self, member in TPUEmbeddingBaseTest