For the custom schedule templates, like the one in:
you can print the schedule (nested for loops) as:
print(tvm.lower(s, arg_bufs, simple_mode=True))
I was wondering if there is a way you can do that for a relay module. For instance after doing:
graph, lib, params = relay.build_module.build(mod, params = dict_params, target=target)
I would really appreciate any information you can provide on this issue.