Searched refs:_create_dense_dataset (Results 1 – 3 of 3) sorted by relevance
79 dataset = self._create_dense_dataset(strategy, include_weights=True)185 dataset = self._create_dense_dataset(strategy)209 dataset = self._create_dense_dataset(strategy)
36 dataset = self._create_dense_dataset(strategy)
305 def _create_dense_dataset(self, strategy, include_weights=False, weight=0.5): member in TPUEmbeddingBaseTest