Searched refs:_create_sparse_dataset (Results 1 – 8 of 8) sorted by relevance
43 dataset = self._create_sparse_dataset(strategy)82 dataset = self._create_sparse_dataset(strategy)115 dataset = self._create_sparse_dataset(strategy)143 dataset = self._create_sparse_dataset(strategy)
45 dataset = self._create_sparse_dataset(strategy)96 sparse = self._create_sparse_dataset(strategy)126 sparse = self._create_sparse_dataset(strategy)176 dataset = self._create_sparse_dataset(strategy, include_weights=True,
36 dataset = self._create_sparse_dataset(strategy)69 dataset = self._create_sparse_dataset(strategy)
37 dataset = self._create_sparse_dataset(strategy)99 input_data = next(iter(self._create_sparse_dataset(strategy)))
124 dataset = self._create_sparse_dataset(strategy)169 input_data = next(iter(self._create_sparse_dataset(strategy)))275 dataset = self._create_sparse_dataset(
102 sparse = self._create_sparse_dataset(strategy, include_weights=True)147 sparse = self._create_sparse_dataset(strategy, include_weights=True)
57 dataset = self._create_sparse_dataset(strategy)
220 def _create_sparse_dataset(self, strategy, include_weights=False, weight=0.5): member in TPUEmbeddingBaseTest