code: def _compile_model(relay_function):
      mod = tvm.IRModule()
      mod["main"] = relay_function
      mod = relay.transform.InferType()(mod)
      tir_mod = _lower_to_tir(mod["main"], _ethos_u55_cascader())[0]
      return tir_mod["main"]
  
file_path: tvm/tests/python/contrib/test_ethosu/cascader/test_integration.py
imports: import pytest
  import numpy
  import tvm
  from tvm import relay
  from tvm.relay.backend.contrib.ethosu.codegen import _create_cascader
  from tvm.relay.backend.contrib.ethosu.tir.compiler import _lower_to_tir
  from tvm.contrib.ethosu.cascader import MemoryRegion, EthosuDeviceConfig
  from ..None import infra
  from .None import infra
label_desc: Use image node_id 2 for calling a global function with example usage: _compile_model(relay_function) and returns: tir_mod