@@ -135,12 +135,15 @@ e.g.
135
135
136
136
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
137
137
138
+ - To include a set of custom operators
138
139
139
- To run with custom torch operators
140
140
.. code-block :: shell
141
- 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
142
141
142
+ 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
143
+
144
+
145
+ - To include a set of custom converters
143
146
144
- To run with custom converters
145
147
.. code-block :: shell
146
- 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
148
+
149
+ 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
0 commit comments