remove read_value

Author: fsx950223

tensorflow_addons/optimizers/gradient_accumulator.py

@@ -82,12 +82,10 @@ def _resource_apply_dense(self, grad, var, apply_state=None):
         def _apply():
             if "apply_state" in self._optimizer._dense_apply_args:
                 train_op = self._optimizer._resource_apply_dense(
-                    accum_gradient.read_value(), var, apply_state=apply_state
+                    accum_gradient, var, apply_state=apply_state
                 )
             else:
-                train_op = self._optimizer._resource_apply_dense(
-                    accum_gradient.read_value(), var
-                )
+                train_op = self._optimizer._resource_apply_dense(accum_gradient, var)
             reset_op = accum_gradient.assign(
                 tf.zeros_like(accum_gradient),
                 use_locking=self._use_locking,
@@ -108,14 +106,14 @@ def _resource_apply_sparse(self, grad: types.TensorLike, var, indices, apply_sta
         def _apply():
             if "apply_state" in self._optimizer._sparse_apply_args:
                 train_op = self._optimizer._resource_apply_sparse(
-                    accum_gradient.read_value(),
+                    accum_gradient,
                     var,
                     indices,
                     apply_state=apply_state,
                 )
             else:
                 train_op = self._optimizer._resource_apply_sparse(
-                    accum_gradient.read_value(), var, indices
+                    accum_gradient, var, indices
                 )
             reset_op = accum_gradient.assign(
                 tf.zeros_like(accum_gradient),