code: def verify_converted_predict_op(self, objax_op, tf_op, shape):
      x1 = np.random.normal(size=shape)
      x2 = np.random.normal(size=shape)
      # due to differences in op implementations, there might be small numerical
      # differences between TF and Objax, thus comparing up to 1e-4 relative tolerance
      np.testing.assert_allclose(
          objax_op(x1),
          tf_op(tf.convert_to_tensor(x1, dtype=tf.float32)),
          rtol=1e-4,
      )
      np.testing.assert_allclose(
          objax_op(x2),
          tf_op(tf.convert_to_tensor(x2, dtype=tf.float32)),
          rtol=1e-4,
      )
  
file_path: objax/tests/objax2tf.py
imports: import shutil
  import tempfile
  import unittest
  import numpy
  import objax
  from objax.zoo.wide_resnet import WideResNet
  import tensorflow
label_desc: Use image node_id 1 for calling the TestObjax2Tf obj's underlying member method code with example usage: obj.verify_converted_predict_op(objax_op, tf_op, shape) without return types