File tree 8 files changed +42
-9
lines changed
8 files changed +42
-9
lines changed Original file line number Diff line number Diff line change 687
687
</ div >
688
688
</ div >
689
689
< p >
690
- < img alt ="cb7dac2a6c864c568c8486f1c561b715 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
690
+ < img alt ="ad355b7bcf1141e383e218be77fc25c8 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
691
691
</ p >
692
692
< h1 id ="notebooks-resnet50-example--page-root ">
693
693
TRTorch Getting Started - ResNet 50
Original file line number Diff line number Diff line change 781
781
</ div >
782
782
</ div >
783
783
< p >
784
- < img alt ="d8c20698903e48c19a5d3dc22f1672d6 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
784
+ < img alt ="089b47097dd24201884cc245f54cbf5e " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
785
785
</ p >
786
786
< h1 id ="notebooks-lenet-getting-started--page-root ">
787
787
TRTorch Getting Started - LeNet
Original file line number Diff line number Diff line change 801
801
</ div >
802
802
</ div >
803
803
< p >
804
- < img alt ="a29c6be5821b4cf79f26897f68580d17 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
804
+ < img alt ="9ed8f5f5946143a6b91c63b8cf43f422 " src ="http://developer.download.nvidia.com/compute/machine-learning/frameworks/nvidia_logo.png "/>
805
805
</ p >
806
806
< h1 id ="notebooks-ssd-object-detection-demo--page-root ">
807
807
Object Detection with TRTorch (SSD)
Original file line number Diff line number Diff line change @@ -11,6 +11,8 @@ trtorch
11
11
Functions
12
12
------------
13
13
14
+ .. autofunction :: set_device
15
+
14
16
.. autofunction :: compile
15
17
16
18
.. autofunction :: convert_method_to_trt_engine
Original file line number Diff line number Diff line change @@ -631,19 +631,24 @@ <h2 id="S">
631
631
< td style ="width: 33%; vertical-align: top; ">
632
632
< ul >
633
633
< li >
634
- < a href ="py_api/logging .html#trtorch.logging.set_is_colored_output_on ">
635
- set_is_colored_output_on () (in module trtorch.logging )
634
+ < a href ="py_api/trtorch .html#trtorch.set_device ">
635
+ set_device () (in module trtorch)
636
636
</ a >
637
637
</ li >
638
638
< li >
639
- < a href ="py_api/logging.html#trtorch.logging.set_logging_prefix ">
640
- set_logging_prefix () (in module trtorch.logging)
639
+ < a href ="py_api/logging.html#trtorch.logging.set_is_colored_output_on ">
640
+ set_is_colored_output_on () (in module trtorch.logging)
641
641
</ a >
642
642
</ li >
643
643
</ ul >
644
644
</ td >
645
645
< td style ="width: 33%; vertical-align: top; ">
646
646
< ul >
647
+ < li >
648
+ < a href ="py_api/logging.html#trtorch.logging.set_logging_prefix ">
649
+ set_logging_prefix() (in module trtorch.logging)
650
+ </ a >
651
+ </ li >
647
652
< li >
648
653
< a href ="py_api/logging.html#trtorch.logging.set_reportable_log_level ">
649
654
set_reportable_log_level() (in module trtorch.logging)
Original file line number Diff line number Diff line change @@ -442,6 +442,32 @@ <h2 id="functions">
442
442
¶
443
443
</ a >
444
444
</ h2 >
445
+ < dl class ="py function ">
446
+ < dt id ="trtorch.set_device ">
447
+ < code class ="sig-prename descclassname ">
448
+ trtorch.
449
+ </ code >
450
+ < code class ="sig-name descname ">
451
+ set_device
452
+ </ code >
453
+ < span class ="sig-paren ">
454
+ (
455
+ </ span >
456
+ < em class ="sig-param ">
457
+ < span class ="n ">
458
+ gpu_id
459
+ </ span >
460
+ </ em >
461
+ < span class ="sig-paren ">
462
+ )
463
+ </ span >
464
+ < a class ="headerlink " href ="#trtorch.set_device " title ="Permalink to this definition ">
465
+ ¶
466
+ </ a >
467
+ </ dt >
468
+ < dd >
469
+ </ dd >
470
+ </ dl >
445
471
< dl class ="py function ">
446
472
< dt id ="trtorch.compile ">
447
473
< code class ="sig-prename descclassname ">
@@ -982,7 +1008,7 @@ <h2 id="functions">
982
1008
< span class ="sig-paren ">
983
1009
)
984
1010
</ span >
985
- → <torch._C.ScriptClass object at 0x7fe4589a4228 >
1011
+ → <torch._C.ScriptClass object at 0x7eff2993c228 >
986
1012
< a class ="headerlink " href ="#trtorch.TensorRTCompileSpec " title ="Permalink to this definition ">
987
1013
¶
988
1014
</ a >
You can’t perform that action at this time.
0 commit comments