code: def testWorksWithFunction(self):
      layer = group_norm.GroupNorm(
          groups=5, create_scale=False, create_offset=False
      )
      function_layer = tf.function(layer)
  
      inputs = tf.ones([2, 3, 3, 10])
      scale = tf.constant(0.5, shape=(10,))
      offset = tf.constant(2.0, shape=(10,))
  
      outputs = layer(inputs, scale, offset)
      function_outputs = function_layer(inputs, scale, offset)
  
      self.assertAllEqual(outputs.numpy(), function_outputs.numpy())
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import group_norm
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 19 for calling the GroupNormTest obj's underlying member method code with example usage: obj.testWorksWithFunction() without return types