code: def testAuxVariablesColocatedWithOriginal(self):
      optimizer = self.make_optimizer(learning_rate=0.1, momentum=0.9)
      if optimizer_tests.is_tf_optimizer(optimizer):
          self.skipTest(
              "TF slot variables are in a different location."
          )
  
      with tf.device("CPU:0"):
          var = tf.Variable(1.0)
      optimizer.apply([tf.constant(0.1)], [var])
      self.assertEqual(
          optimizer.accumulated_momentum[0].device, var.device
      )
  
imports: from sonnet.src import test_utils
  from sonnet.src.optimizers import momentum
  from sonnet.src.optimizers import optimizer_tests
  import tensorflow
label_desc: Use image node_id 8 for calling the MomentumTest obj's underlying member method code with example usage: obj.testAuxVariablesColocatedWithOriginal() without return types