Skip to content

Commit a509bda

Browse files
author
TRTorch Github Bot
committed
docs: [Automated] Regenerating documenation for e5e9620
Signed-off-by: TRTorch Github Bot <[email protected]>
1 parent e5e9620 commit a509bda

File tree

5 files changed

+13
-8
lines changed

5 files changed

+13
-8
lines changed

docs/_notebooks/Resnet50-example.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -690,7 +690,7 @@
690690
</div>
691691
</div>
692692
<p>
693-
<img alt="c85e3eee5f964c87bdbd3da68be9c4e6" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
693+
<img alt="220e18e0c7ee47e29626163591408bba" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
694694
</p>
695695
<h1 id="notebooks-resnet50-example--page-root">
696696
TRTorch Getting Started - ResNet 50

docs/_notebooks/lenet-getting-started.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -784,7 +784,7 @@
784784
</div>
785785
</div>
786786
<p>
787-
<img alt="48e6dc90714648bba0dd002adedc7a38" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
787+
<img alt="9df4c63d2f1d42129c0898d6cc79d912" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
788788
</p>
789789
<h1 id="notebooks-lenet-getting-started--page-root">
790790
TRTorch Getting Started - LeNet

docs/_notebooks/ssd-object-detection-demo.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -804,7 +804,7 @@
804804
</div>
805805
</div>
806806
<p>
807-
<img alt="8fef448da1fa4681b0f1f1027eb1c9a7" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
807+
<img alt="13aaf97442cb41d498fac3a7296402fd" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
808808
</p>
809809
<h1 id="notebooks-ssd-object-detection-demo--page-root">
810810
Object Detection with TRTorch (SSD)

docs/py_api/trtorch.html

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -596,11 +596,11 @@ <h2 id="functions">
596596
<span class="s2">"workspace_size"</span><span class="p">:</span> <span class="mi">0</span><span class="p">,</span> <span class="c1"># Maximum size of workspace given to TensorRT</span>
597597
<span class="s2">"max_batch_size"</span><span class="p">:</span> <span class="mi">0</span><span class="p">,</span> <span class="c1"># Maximum batch size (must be &gt;= 1 to be set, 0 means not set)</span>
598598
<span class="s2">"torch_fallback"</span><span class="p">:</span> <span class="p">{</span>
599-
<span class="s2">"enabled"</span><span class="p">:</span> <span class="kc">True</span><span class="p">,</span>
599+
<span class="s2">"enabled"</span><span class="p">:</span> <span class="kc">True</span><span class="p">,</span> <span class="c1"># Turn on or turn off falling back to PyTorch if operations are not supported in TensorRT</span>
600600
<span class="s2">"force_fallback_ops"</span><span class="p">:</span> <span class="p">[</span>
601-
<span class="s2">"aten::max_pool2d"</span>
601+
<span class="s2">"aten::max_pool2d"</span> <span class="c1"># List of specific ops to require running in PyTorch</span>
602602
<span class="p">],</span>
603-
<span class="s2">"min_block_size"</span><span class="p">:</span> <span class="mi">1</span>
603+
<span class="s2">"min_block_size"</span><span class="p">:</span> <span class="mi">3</span> <span class="c1"># Minimum number of ops an engine must incapsulate to be run in TensorRT</span>
604604
<span class="p">}</span>
605605
<span class="p">}</span>
606606
</pre>
@@ -1018,7 +1018,7 @@ <h2 id="functions">
10181018
<span class="sig-paren">
10191019
)
10201020
</span>
1021-
→ &lt;torch._C.ScriptClass object at 0x7fabde990ef0&gt;
1021+
→ &lt;torch._C.ScriptClass object at 0x7fdd093be6f0&gt;
10221022
<a class="headerlink" href="#trtorch.TensorRTCompileSpec" title="Permalink to this definition">
10231023
10241024
</a>
@@ -1052,6 +1052,11 @@ <h2 id="functions">
10521052
</code>
10531053
, describing the input sizes or ranges for inputs
10541054
to the graph. All other keys are optional. Entries for each method to be compiled.
1055+
</p>
1056+
<p>
1057+
Note: Partial compilation of TorchScript modules is not supported through the PyTorch TensorRT backend
1058+
If you need this feature, use trtorch.compile to compile your module. Usage of the resulting module is
1059+
as if you were using the TensorRT integration.
10551060
</p>
10561061
<div class="highlight-py notranslate">
10571062
<div class="highlight">

docs/searchindex.js

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)