Searched refs:_decay_weights_sparse_op (Results 1 – 1 of 1) sorted by relevance
179 def _decay_weights_sparse_op(self, var, indices, scatter_add): member in DecoupledWeightDecayExtension198 decay_op = self._decay_weights_sparse_op(var, grad.indices, scatter_add)212 decay_op = self._decay_weights_sparse_op(var, indices, scatter_add)