code: def testConv(self):
      y = extensions.conv(
          np.ones([5, 320, 480, 3], dtype=np.float32),
          np.ones([3, 4, 3, 11], dtype=np.float32),
          [1, 1],
          "SAME",
          ("NHWC", "HWIO", "NHWC"),
      )
      self.assertAllClose(y.shape, [5, 320, 480, 11])
      self.assertAllClose(
          y,
          tf.nn.conv2d(
              input=tf.ones([5, 320, 480, 3], dtype=tf.float32),
              filters=tf.ones([3, 4, 3, 11], dtype=tf.float32),
              strides=1,
              padding="SAME",
          ),
      )
  
imports: import functools
  import itertools
  from absl import flags
  from absl.testing import parameterized
  from jax import lax
  import numpy
  import tensorflow.compat.v2
  from trax.tf_numpy import extensions
  import trax.tf_numpy.numpy
label_desc: Use image node_id 18 for calling the ExtensionsTest obj's underlying member method code with example usage: obj.testConv() without return types