feat(//py): register trtorch with torch op library to support

deserializing modules in python

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

Diff for: py/trtorch/__init__.py

@@ -11,3 +11,7 @@
 from trtorch._compiler import *
 from trtorch._types import *
 from trtorch import logging
+
+def _register_with_torch():
+    trtorch_dir = os.path.dirname(__file__)
+    torch.ops.load_library(trtorch_dir + '/lib/trtorch.so')