(self, grad, var)
| 147 | self._epsilon_t = ops.convert_to_tensor(epsilon, name="epsilon") |
| 148 | |
| 149 | def _apply_dense(self, grad, var): |
| 150 | m = self.get_slot(var, "m") |
| 151 | v = self.get_slot(var, "v") |
| 152 | beta1_power, beta2_power = self._get_beta_accumulators() |
| 153 | return training_ops.apply_adam( |
| 154 | var, |
| 155 | m, |
| 156 | v, |
| 157 | math_ops.cast(beta1_power, var.dtype.base_dtype), |
| 158 | math_ops.cast(beta2_power, var.dtype.base_dtype), |
| 159 | math_ops.cast(self._lr_t, var.dtype.base_dtype), |
| 160 | math_ops.cast(self._beta1_t, var.dtype.base_dtype), |
| 161 | math_ops.cast(self._beta2_t, var.dtype.base_dtype), |
| 162 | math_ops.cast(self._epsilon_t, var.dtype.base_dtype), |
| 163 | grad, |
| 164 | use_locking=self._use_locking).op |
| 165 | |
| 166 | def _resource_apply_dense(self, grad, var): |
| 167 | m = self.get_slot(var, "m") |
nothing calls this directly
no test coverage detected