Many thanks @comaniac. It worked. Also, I had to fix graph_runtime error mentioned here AttributeError: module ‘tvm.contrib.graph_runtime’ has no attribute 'GraphModule - Questions - Apache TVM Discuss.
Please let me know if I can refer to these changes listed somewhere?
Thanks.