def METHOD_NAME(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"]