Skip to content

Commit d345407

Browse files
committed
Merge branch 'master' into resolve_inputs_fix
2 parents 8d59343 + 359d07a commit d345407

10 files changed

+43
-10
lines changed

docs/_cpp_api/function_torch__tensorrt_8h_1a710df824a7718b440e4bc17bf9693cef.html

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -635,6 +635,21 @@ <h2 id="function-documentation">
635635
<p>
636636
Takes a pre-built serialized TensorRT engine and embeds it in a TorchScript module. Registers execution of the engine as the forward method of the module Forward is defined as: forward(Tensor[]) -&gt; Tensor[]
637637
</p>
638+
<p>
639+
TensorRT bindings must have names with the following format:
640+
</p>
641+
<ul class="simple">
642+
<li>
643+
<p>
644+
[symbol].[index in input / output array] ex.
645+
</p>
646+
</li>
647+
<li>
648+
<p>
649+
[x.0, x.1, x.2] -&gt; [y.0]
650+
</p>
651+
</li>
652+
</ul>
638653
<dl class="field-list simple">
639654
<dt class="field-odd">
640655
Parameters
@@ -670,7 +685,7 @@ <h2 id="function-documentation">
670685
</dt>
671686
<dd class="field-even">
672687
<p>
673-
: A new module trageting a TensorRT engine
688+
: A new module targeting a TensorRT engine
674689
</p>
675690
</dd>
676691
</dl>

docs/_notebooks/CitriNet-example.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -857,7 +857,7 @@
857857
</div>
858858
</div>
859859
<p>
860-
<img alt="20d57019de6243a68e76654f4ea8f8ae" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
860+
<img alt="c61fa285604242db814bfb71fcb3f44a" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
861861
</p>
862862
<section id="Torch-TensorRT-Getting-Started---CitriNet">
863863
<h1 id="notebooks-citrinet-example--page-root">

docs/_notebooks/EfficientNet-example.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -857,7 +857,7 @@
857857
</div>
858858
</div>
859859
<p>
860-
<img alt="8ef80a67a46b4935a6252c9d572a57b8" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
860+
<img alt="1b913ec657a24201a6c8f2c361875572" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
861861
</p>
862862
<section id="Torch-TensorRT-Getting-Started---EfficientNet-B0">
863863
<h1 id="notebooks-efficientnet-example--page-root">

docs/_notebooks/Hugging-Face-BERT.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -809,7 +809,7 @@
809809
</div>
810810
</div>
811811
<p>
812-
<img alt="2c52902157e54c8e8068349075e9eeff" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_resnet_notebook.png"/>
812+
<img alt="9789ac3c1cdf4cef840d6171a4052c75" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_resnet_notebook.png"/>
813813
</p>
814814
<section id="Masked-Language-Modeling-(MLM)-with-Hugging-Face-BERT-Transformer">
815815
<h1 id="notebooks-hugging-face-bert--page-root">

docs/_notebooks/Resnet50-example.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -857,7 +857,7 @@
857857
</div>
858858
</div>
859859
<p>
860-
<img alt="9cbd1c2fc7c541b5b2afa929886f578b" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_resnet_notebook.png"/>
860+
<img alt="a1c22a88cb274311833320666e204a51" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_resnet_notebook.png"/>
861861
</p>
862862
<section id="Torch-TensorRT-Getting-Started---ResNet-50">
863863
<h1 id="notebooks-resnet50-example--page-root">

docs/_notebooks/lenet-getting-started.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -895,7 +895,7 @@
895895
</div>
896896
</div>
897897
<p>
898-
<img alt="ff2c118facde47ba9f8b3b3129d8fb55" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
898+
<img alt="3f6a6bf3756a4dc996a1bfeea83e15e1" src="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png"/>
899899
</p>
900900
<section id="Torch-TensorRT-Getting-Started---LeNet">
901901
<h1 id="notebooks-lenet-getting-started--page-root">

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -915,7 +915,7 @@
915915
</div>
916916
</div>
917917
<p>
918-
<img alt="76b4640a03564b6c8e5f8ec237a17f05" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_ssd_notebook.png"/>
918+
<img alt="aac56f28a5e54218ae3d219392f48c65" src="https://developer.download.nvidia.com/tesla/notebook_assets/nv_logo_torch_trt_ssd_notebook.png"/>
919919
</p>
920920
<section id="Object-Detection-with-Torch-TensorRT-(SSD)">
921921
<h1 id="notebooks-ssd-object-detection-demo--page-root">

docs/_notebooks/vgg-qat.html

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -922,7 +922,7 @@ <h2 id="Overview">
922922
</div>
923923
<p>
924924
## 2. VGG16 Overview ### Very Deep Convolutional Networks for Large-Scale Image Recognition VGG is one of the earliest family of image classification networks that first used small (3x3) convolution filters and achieved significant improvements on ImageNet recognition challenge. The network architecture looks as follows
925-
<img alt="c96789a7ffcd49a291298042bad651d4" src="https://neurohive.io/wp-content/uploads/2018/11/vgg16-1-e1542731207177.png"/>
925+
<img alt="65356fb264d64a9a9295dea65f568939" src="https://neurohive.io/wp-content/uploads/2018/11/vgg16-1-e1542731207177.png"/>
926926
</p>
927927
<p>
928928
## 3. Training a baseline VGG16 model We train VGG16 on CIFAR10 dataset. Define training and testing datasets and dataloaders. This will download the CIFAR 10 data in your

docs/py_api/ts.html

Lines changed: 19 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1851,6 +1851,24 @@ <h2 id="functions">
18511851
</p>
18521852
</div>
18531853
</blockquote>
1854+
<dl>
1855+
<dt>
1856+
TensorRT bindings must have names with the following format:
1857+
</dt>
1858+
<dd>
1859+
<ul class="simple">
1860+
<li>
1861+
<p>
1862+
[symbol].[index in input / output array]
1863+
</p>
1864+
</li>
1865+
</ul>
1866+
<p>
1867+
ex.
1868+
- [x.0, x.1, x.2] -&gt; [y.0]
1869+
</p>
1870+
</dd>
1871+
</dl>
18541872
<p>
18551873
Module can be save with engine embedded with torch.jit.save and moved / loaded according to torch_tensorrt portability rules
18561874
</p>
@@ -2066,7 +2084,7 @@ <h2 id="functions">
20662084
at
20672085
</span>
20682086
<span class="pre">
2069-
0x7f97985ef6f0&gt;
2087+
0x7fdeab7877b0&gt;
20702088
</span>
20712089
</span>
20722090
</span>

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)