code: def testAuxVariablesColocatedWithOriginal(self):
      optimizer = self.make_optimizer(learning_rate=0.001)
      with tf.device("CPU:0"):
          var = tf.Variable(1.0)
      optimizer.apply([tf.constant(0.1)], [var])
      self.assertEqual(optimizer.m[0].device, var.device)
      self.assertEqual(optimizer.v[0].device, var.device)
  
file_path: sonnet/sonnet/src/optimizers/adam_test.py
imports: from sonnet.src import test_utils
  from sonnet.src.optimizers import adam
  from sonnet.src.optimizers import optimizer_tests
  import tensorflow
label_desc: Use image node_id 6 for calling the AdamTest obj's underlying member method code with example usage: obj.testAuxVariablesColocatedWithOriginal() without return types