Searched refs:_ShardTestEmbeddings (Results 1 – 1 of 1) sorted by relevance
611 def _ShardTestEmbeddings(self, weights, biases, num_shards): member in ComputeSampledLogitsTest806 weight_shards, bias_shards = self._ShardTestEmbeddings(866 weight_shards, bias_shards = self._ShardTestEmbeddings(922 weight_shards, bias_shards = self._ShardTestEmbeddings(