Searched refs:_random_weights (Results 1 – 2 of 2) sorted by relevance
47 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): member in SafeEmbeddingLookupSparseTest117 embedding_weights = self._random_weights()130 embedding_weights = self._random_weights()144 embedding_weights = self._random_weights()158 embedding_weights = self._random_weights(num_shards=3)172 embedding_weights = self._random_weights(num_shards=3)187 embedding_weights = self._random_weights()200 embedding_weights = self._random_weights()216 embedding_weights = self._random_weights()232 embedding_weights = self._random_weights(num_shards=3)[all …]
792 def _random_weights(self, vocab_size=4, embed_dim=4, num_shards=1): member in SafeEmbeddingLookupSparseTest863 embedding_weights = self._random_weights()878 embedding_weights = self._random_weights()895 embedding_weights = self._random_weights()911 embedding_weights = self._random_weights(num_shards=3)927 embedding_weights = self._random_weights(num_shards=3)943 embedding_weights = self._random_weights()958 embedding_weights = self._random_weights()977 embedding_weights = self._random_weights()995 embedding_weights = self._random_weights(num_shards=3)[all …]