code: def test_squeeze(run_module):
      def get_graph(x_shape, axis, dtype):
          x = relay.var("x", shape=(x_shape), dtype=dtype)
          out = relay.squeeze(x, axis=axis)
          f = relay.Function([x], out)
          return f, {"x": x_shape}, []
  
      for dtype in SUPPORTED_DTYPES:
          run_and_verify_func(
              get_graph((1, 5, 1, 1), (2, 3), dtype=dtype),
              run_module=run_module,
              data_type=dtype,
          )
          run_and_verify_func(
              get_graph((1, 3, 1), (-1,), dtype=dtype),
              run_module=run_module,
              data_type=dtype,
          )
  
imports: import itertools
  import logging
  from typing import Tuple
  import numpy
  import pytest
  import tvm
  import tvm.relay.testing
  import tvm.testing
  from tvm import relay
  from tvm.contrib.download import download
  from tvm.relay import Any, GlobalVar
  from tvm.relay.expr_functor import ExprVisitor
  from tvm.relay.op.contrib import tensorrt
label_desc: Use image node_id 21 for calling a global function with example usage: test_squeeze(run_module) and returns: f,  dict,  list