Searched refs:_ShardTestEmbeddings (Results 1 – 1 of 1) sorted by relevance
566 def _ShardTestEmbeddings(self, weights, biases, num_shards): member in ComputeSampledLogitsTest761 weight_shards, bias_shards = self._ShardTestEmbeddings(821 weight_shards, bias_shards = self._ShardTestEmbeddings(877 weight_shards, bias_shards = self._ShardTestEmbeddings(