code: def testFunction(self, with_bias):
      linear_1 = linear.Linear(
          3, with_bias=with_bias, w_init=tf.ones_initializer()
      )
      linear_2 = linear.Linear(
          3, with_bias=with_bias, w_init=tf.ones_initializer()
      )
      defun_linear = tf.function(linear_2)
  
      iterations = 5
  
      for _ in range(iterations):
          x = tf.random.uniform([1, 5])
          y1 = linear_1(x)
          y2 = defun_linear(x)
  
          self.assertAllClose(
              self.evaluate(y1), self.evaluate(y2), atol=1e-4
          )
  
imports: from absl.testing import parameterized
  import numpy
  from sonnet.src import linear
  from sonnet.src import test_utils
  import tensorflow
label_desc: Use image node_id 10 for calling the LinearTest obj's underlying member method code with example usage: obj.testFunction(with_bias) without return types