code: def infer_shape(expr):
      return relay.transform.InferType()(tvm.IRModule.from_expr(expr))[
          "main"
      ].body.checked_type.shape
  
imports: import numpy
  import tvm
  from tvm import relay
  from tvm.ir import TensorAffineType, TupleAffineType
  from tvm.relay.qnn.op import canonicalizations
  from tvm.tir import bijective_layout
  from ..op import register_fake_quantization_to_integer, register_optional_fake_quantization_to_integer
label_desc: Use image node_id 3 for calling a global function with example usage: infer_shape(expr) and returns: relay