Update torchtrtc.rst

pytorch · Aug 25, 2022 · d43606b · d43606b
1 parent 5d1acba
commit d43606b
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/docsrc/cli/torchtrtc.rst b/docsrc/cli/torchtrtc.rst
@@ -135,12 +135,15 @@ e.g.
 
     torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@f16%contiguous" -p f16
 
+- To include a set of custom operators
 
-To run with custom torch operators
 .. code-block:: shell
-torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-torch-ops=<path to custom library> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
 
+    torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-torch-ops=<path to custom library .so file> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
+
+
+- To include a set of custom converters
 
-To run with custom converters
 .. code-block:: shell
-torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-converters=<path to custom library> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16
+
+    torchtrtc tests/modules/ssd_traced.jit.pt ssd_trt.ts --custom-converters=<path to custom library .so file> "[(1,3,300,300); (1,3,512,512); (1, 3, 1024, 1024)]@fp16%contiguous" -p f16