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)
Exemple #2
0
 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)
Exemple #3
0
 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)
Exemple #4
0
 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)