Searched refs:_mod_sharded_biases (Results 1 – 1 of 1) sorted by relevance
123 def _mod_sharded_biases(self): member in RankSampledSoftmaxLossTest250 self._testCompareWithNN(self._mod_sharded_weights, self._mod_sharded_biases,