code: def get_beta_accumulators(opt, dtype):
      local_step = tf.cast(opt.iterations + 1, dtype)
      beta_1_t = tf.cast(opt._get_hyper("beta_1"), dtype)
      beta_1_power = tf.math.pow(beta_1_t, local_step)
      beta_2_t = tf.cast(opt._get_hyper("beta_2"), dtype)
      beta_2_power = tf.math.pow(beta_2_t, local_step)
      return (beta_1_power, beta_2_power)
  
file_path: addons/tensorflow_addons/optimizers/tests/lazy_adam_test.py
imports: import numpy
  import tensorflow
  from tensorflow_addons.optimizers import lazy_adam
  from tensorflow_addons.utils import test_utils
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: get_beta_accumulators(opt, dtype) and returns: beta_1_power,  beta_2_power