(self, grad, var, indices, scatter_add)
| 209 | ''' |
| 210 | |
| 211 | def _apply_sparse_shared(self, grad, var, indices, scatter_add): |
| 212 | m = self.get_slot(var, 'm') |
| 213 | v = self.get_slot(var, 'v') |
| 214 | beta1_power, beta2_power = self._get_beta_accumulators() |
| 215 | return training_ops.sparse_apply_adam( |
| 216 | var, m, v, |
| 217 | math_ops.cast(beta1_power, var.dtype.base_dtype), |
| 218 | math_ops.cast(beta2_power, var.dtype.base_dtype), |
| 219 | math_ops.cast(self._lr_t, var.dtype.base_dtype), |
| 220 | math_ops.cast(self._beta1_t, var.dtype.base_dtype), |
| 221 | math_ops.cast(self._beta2_t, var.dtype.base_dtype), |
| 222 | math_ops.cast(self._epsilon_t, var.dtype.base_dtype), |
| 223 | grad, indices, use_locking=self._use_locking) |
| 224 | |
| 225 | def _resource_apply_sparse_shared(self, grad, var, indices, scatter_add): |
| 226 | m = self.get_slot(var, 'm') |
no test coverage detected