TVM Runtime - Print Intermediate Outputs

You can use follow snippet to load data from output_tensors.params into memory and then proceed analysis or dump in more suitable foprm

data = relay.load_param_dict(bytearray(open("./_tvmdbg_device_CPU_0/output_tensors.params", "rb").read()))

data will contain a map of output names to NDArray

1 Like