Ejemplo n.º 1
0
 def _resource_apply_sparse(self, grad, var, indices):
     return training_ops.resource_sparse_apply_proximal_gradient_descent(
         var.handle,
         math_ops.cast(self._learning_rate_tensor, grad.dtype),
         math_ops.cast(self._l1_regularization_strength_tensor, grad.dtype),
         math_ops.cast(self._l2_regularization_strength_tensor, grad.dtype),
         grad,
         indices,
         use_locking=self._use_locking)
 def _resource_apply_sparse(self, grad, var, indices):
   return training_ops.resource_sparse_apply_proximal_gradient_descent(
       var.handle,
       math_ops.cast(self._learning_rate_tensor, grad.dtype),
       math_ops.cast(self._l1_regularization_strength_tensor, grad.dtype),
       math_ops.cast(self._l2_regularization_strength_tensor, grad.dtype),
       grad,
       indices,
       use_locking=self._use_locking)