code: def test_tf_function(self, resnet_v2):
      image = tf.random.normal([2, 64, 64, 3])
      model = resnet.ResNet(
          [1, 1, 1, 1],
          10,
          resnet_v2=resnet_v2,
      )
      f = tf.function(model)
  
      logits = f(image, is_training=True)
      self.assertIsNotNone(logits)
      self.assertEqual(logits.shape, [2, 10])
      self.assertAllEqual(
          model(image, is_training=True).numpy(), logits.numpy()
      )
  
file_path: sonnet/sonnet/src/nets/resnet_test.py
imports: from absl.testing import parameterized
  from sonnet.src import test_utils
  from sonnet.src.nets import resnet
  import tensorflow
label_desc: Use image node_id 2 for calling the ResnetTest obj's underlying member method code with example usage: obj.test_tf_function(resnet_v2) without return types