Lines Matching refs:grad
34 def _apply_dense(self, grad, var):
48 grad,
52 def _resource_apply_dense(self, grad, var):
60 math_ops.cast(beta1_power, grad.dtype.base_dtype),
61 math_ops.cast(beta2_power, grad.dtype.base_dtype),
62 math_ops.cast(self._lr_t, grad.dtype.base_dtype),
63 math_ops.cast(self._beta1_t, grad.dtype.base_dtype),
64 math_ops.cast(self._beta2_t, grad.dtype.base_dtype),
65 math_ops.cast(self._epsilon_t, grad.dtype.base_dtype),
66 grad,
70 def _apply_sparse_shared(self, grad, var, indices, scatter_add):
81 m_scaled_g_values = grad * (1 - beta1_t)
89 v_scaled_g_values = (grad * grad) * (1 - beta2_t)