(self, grad, var)
| 164 | use_locking=self._use_locking).op |
| 165 | |
| 166 | def _resource_apply_dense(self, grad, var): |
| 167 | m = self.get_slot(var, "m") |
| 168 | v = self.get_slot(var, "v") |
| 169 | beta1_power, beta2_power = self._get_beta_accumulators() |
| 170 | return training_ops.resource_apply_adam( |
| 171 | var.handle, |
| 172 | m.handle, |
| 173 | v.handle, |
| 174 | math_ops.cast(beta1_power, grad.dtype.base_dtype), |
| 175 | math_ops.cast(beta2_power, grad.dtype.base_dtype), |
| 176 | math_ops.cast(self._lr_t, grad.dtype.base_dtype), |
| 177 | math_ops.cast(self._beta1_t, grad.dtype.base_dtype), |
| 178 | math_ops.cast(self._beta2_t, grad.dtype.base_dtype), |
| 179 | math_ops.cast(self._epsilon_t, grad.dtype.base_dtype), |
| 180 | grad, |
| 181 | use_locking=self._use_locking) |
| 182 | |
| 183 | ''' |
| 184 | def _apply_sparse_shared(self, grad, var, indices, scatter_add): |
nothing calls this directly
no test coverage detected