Searched refs:_mod_sharded_weights (Results 1 – 1 of 1) sorted by relevance
86 def _mod_sharded_weights(self): member in RankSampledSoftmaxLossTest247 self._testCompareWithNN(self._mod_sharded_weights, self._biases, 'mod')250 self._testCompareWithNN(self._mod_sharded_weights, self._mod_sharded_biases,