code: def testWithTFFunction(self, autograph):
      ema_1 = moving_averages.ExponentialMovingAverage(0.95)
      ema_2 = moving_averages.ExponentialMovingAverage(0.95)
      ema_func = tf.function(ema_2, autograph=autograph)
  
      for _ in range(10):
          x = tf.random.uniform((), 0, 10)
          self.assertAllClose(
              ema_1(x).numpy(),
              ema_func(x).numpy(),
              atol=1e-3,
              rtol=1e-5,
          )
  
file_path: sonnet/sonnet/src/moving_averages_test.py
imports: from absl.testing import parameterized
  from sonnet.src import moving_averages
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 6 for calling the ExponentialMovingAverageTest obj's underlying member method code with example usage: obj.testWithTFFunction(autograph) without return types