convert torch module to tensorrt network or tvm function - View it on GitHub