Skip to content

Commit 4d8b299

Browse files
authored
Update use_from_pytorch.rst
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 20022d4 commit 4d8b299

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docsrc/tutorials/use_from_pytorch.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ Start by loading ``trtorch`` into your application.
1414
import trtorch
1515
1616
17-
Then given a TorchScript module, you can lower it to TensorRT using the ``torch._C._jit_to_tensorrt`` API.
17+
Then given a TorchScript module, you can compile it with TensorRT using the ``torch._C._jit_to_backend("tensorrt", ...)`` API.
1818

1919
.. code-block:: python
2020
@@ -52,7 +52,7 @@ at the documentation for the TRTorch ``TensorRTCompileSpec`` API.
5252
})
5353
}
5454
55-
Now to compile with TRTorch, provide the target module objects and the spec dictionary to ``torch._C._jit_to_tensorrt``
55+
Now to compile with TRTorch, provide the target module objects and the spec dictionary to ``torch._C._jit_to_backend("tensorrt", ...)``
5656

5757
.. code-block:: python
5858

0 commit comments

Comments
 (0)