How to get the schedule information of OPs in relay?

Questions
2
  • graph: str
  • mod: tvm.relay.module.Module
  • params: dict

As TVM only takes tvm.Tensor / tvm as input, I tried to fetch the computed result

ctx = tvm.gpu()
tvm_module = tvm.contrib.graph_runtime.create(graph, lib, ctx)
tvm_module.set_input("data", dshape)
tvm_module.set_input(**params)
tvm_module.run()
tvm_output = tvm_module.get_output(0)

But the tvm_output returns type tvm.ndarray.NDArray.