def METHOD_NAME(self, grads_and_vars):
    dtype = self._u_product.dtype
    self._u_product.assign(
        self._u_product
        * self.beta_1
        * (
            1.0
            - 0.5 * ops.power(0.96, ops.cast(self.iterations + 1, dtype))
        )
    )
    super().METHOD_NAME(grads_and_vars)