Skip to content

Commit 03a6ca4

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

File tree

8 files changed

+97
-7
lines changed

8 files changed

+97
-7
lines changed

docs/_notebooks/Resnet50-example.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -693,7 +693,7 @@
693693
</div>
694694
</div>
695695
<p>
696-
<img alt="a8ae5e030e6848f5af796036d2f76e80" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
696+
<img alt="144d9f5c2b484f63937e6da2ba926676" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
697697
</p>
698698
<h1 id="notebooks-resnet50-example--page-root">
699699
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
@@ -787,7 +787,7 @@
787787
</div>
788788
</div>
789789
<p>
790-
<img alt="2c7d2affdb8e4a87ae75143ff13ecf74" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
790+
<img alt="40e215c8544444dba32c7ee842b4df84" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
791791
</p>
792792
<h1 id="notebooks-lenet-getting-started--page-root">
793793
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
@@ -807,7 +807,7 @@
807807
</div>
808808
</div>
809809
<p>
810-
<img alt="9f0fb1d37105425ebbf46f3a550a70f1" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
810+
<img alt="965dc6a6673e4ac5b13efff3572f46ab" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
811811
</p>
812812
<h1 id="notebooks-ssd-object-detection-demo--page-root">
813813
Object Detection with TRTorch (SSD)

docs/_sources/py_api/trtorch.rst.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,8 @@ Functions
1919

2020
.. autofunction:: check_method_op_support
2121

22+
.. autofunction:: embed_engine_in_new_module
23+
2224
.. autofunction:: get_build_info
2325

2426
.. autofunction:: dump_build_info

docs/genindex.html

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -501,14 +501,19 @@ <h2 id="E">
501501
<td style="width: 33%; vertical-align: top;">
502502
<ul>
503503
<li>
504-
<a href="py_api/trtorch.html#trtorch.EngineCapability">
505-
EngineCapability (class in trtorch)
504+
<a href="py_api/trtorch.html#trtorch.embed_engine_in_new_module">
505+
embed_engine_in_new_module() (in module trtorch)
506506
</a>
507507
</li>
508508
</ul>
509509
</td>
510510
<td style="width: 33%; vertical-align: top;">
511511
<ul>
512+
<li>
513+
<a href="py_api/trtorch.html#trtorch.EngineCapability">
514+
EngineCapability (class in trtorch)
515+
</a>
516+
</li>
512517
<li>
513518
<a href="py_api/logging.html#trtorch.logging.Level.Error">
514519
Error (trtorch.logging.Level attribute)

docs/objects.inv

38 Bytes
Binary file not shown.

docs/py_api/trtorch.html

Lines changed: 84 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -918,6 +918,89 @@ <h2 id="functions">
918918
</dl>
919919
</dd>
920920
</dl>
921+
<dl class="py function">
922+
<dt id="trtorch.embed_engine_in_new_module">
923+
<code class="sig-prename descclassname">
924+
trtorch.
925+
</code>
926+
<code class="sig-name descname">
927+
embed_engine_in_new_module
928+
</code>
929+
<span class="sig-paren">
930+
(
931+
</span>
932+
<em class="sig-param">
933+
<span class="n">
934+
serialized_engine
935+
</span>
936+
<span class="p">
937+
:
938+
</span>
939+
<span class="n">
940+
bytes
941+
</span>
942+
</em>
943+
<span class="sig-paren">
944+
)
945+
</span>
946+
→ torch.jit._script.ScriptModule
947+
<a class="headerlink" href="#trtorch.embed_engine_in_new_module" title="Permalink to this definition">
948+
949+
</a>
950+
</dt>
951+
<dd>
952+
<p>
953+
Takes a pre-built serialized TensorRT engine and embeds it within a TorchScript module
954+
</p>
955+
<p>
956+
Takes a pre-built serialied TensorRT engine (as bytes) and embeds it within a TorchScript module.
957+
Registers the forward method to execute the TensorRT engine with the function signature:
958+
</p>
959+
<blockquote>
960+
<div>
961+
<p>
962+
forward(Tensor[]) -&gt; Tensor[]
963+
</p>
964+
</div>
965+
</blockquote>
966+
<p>
967+
Module can be save with engine embedded with torch.jit.save and moved / loaded according to TRTorch portability rules
968+
</p>
969+
<dl class="field-list simple">
970+
<dt class="field-odd">
971+
Parameters
972+
</dt>
973+
<dd class="field-odd">
974+
<p>
975+
<strong>
976+
serialized_engine
977+
</strong>
978+
(
979+
<em>
980+
bytes
981+
</em>
982+
) – Serialized TensorRT engine from either TRTorch or TensorRT APIs
983+
</p>
984+
</dd>
985+
<dt class="field-even">
986+
Returns
987+
</dt>
988+
<dd class="field-even">
989+
<p>
990+
New TorchScript module with engine embedded
991+
</p>
992+
</dd>
993+
<dt class="field-odd">
994+
Return type
995+
</dt>
996+
<dd class="field-odd">
997+
<p>
998+
torch.jit.ScriptModule
999+
</p>
1000+
</dd>
1001+
</dl>
1002+
</dd>
1003+
</dl>
9211004
<dl class="py function">
9221005
<dt id="trtorch.get_build_info">
9231006
<code class="sig-prename descclassname">
@@ -1021,7 +1104,7 @@ <h2 id="functions">
10211104
<span class="sig-paren">
10221105
)
10231106
</span>
1024-
→ &lt;torch._C.ScriptClass object at 0x7ff6a30319b0&gt;
1107+
→ &lt;torch._C.ScriptClass object at 0x7f35105b3e70&gt;
10251108
<a class="headerlink" href="#trtorch.TensorRTCompileSpec" title="Permalink to this definition">
10261109
10271110
</a>

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)