def _resource_apply_dense(self, grad, var): acc = self.get_slot(var, "accumulator") return training_ops.resource_apply_adagrad( var.handle, acc.handle, math_ops.cast(self._learning_rate_tensor, grad.dtype.base_dtype), grad, use_locking=self._use_locking)
def _resource_apply_dense(self, grad, var, state): acc = state.get_slot(var, "accumulator") return training_ops.resource_apply_adagrad( var.handle, acc.handle, state.get_hyper("learning_rate", var.dtype.base_dtype), grad, use_locking=self._use_locking)