code: def testInstanceNormCorrectAxis(self):
      layer = axis_norm.InstanceNorm(
          create_scale=True, create_offset=True
      )
  
      inputs = tf.ones([3, 4, 5, 6])
      layer(inputs)
  
      self.assertEqual(layer._axis, (1, 2))
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import axis_norm
  from sonnet.src import initializers
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 22 for calling the LayerNormTest obj's underlying member method code with example usage: obj.testInstanceNormCorrectAxis() without return types