Searched refs:_div_sharded_weights (Results 1 – 1 of 1) sorted by relevance
62 def _div_sharded_weights(self): member in RankSampledSoftmaxLossTest240 self._testCompareWithNN(self._div_sharded_weights, self._biases, 'div')243 self._testCompareWithNN(self._div_sharded_weights, self._div_sharded_biases,