Skip to content

Pull requests: NVIDIA-AI-IOT/torch2trt

New pull request New
50 Open 164 Closed
50 Open 164 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

input of batchnorm1d can also be (N,C)
#934 opened Jun 17, 2024 by taotaoorange Loading…
Support text models
#932 opened Jun 8, 2024 by aliencaocao Loading…
1
Fixed TEST_OUTPUT.md table report (PSNR and MSE were not included in the table header)
#851 opened Feb 21, 2023 by manuelrames Loading…
[torch2trt/torch2trt.py] Address issue #848: Add automatic type promotion when comparing tensors of different dtypes.
#849 opened Feb 15, 2023 by chaoz-dev Loading…
1
[bug-fix] Remove assert-invalidating parentheses
#837 opened Feb 3, 2023 by NellyWhads Loading…
2
[converters/stack.py] Update stack converter to support dynamic shapes. Resolves #830.
#831 opened Jan 24, 2023 by chaoz-dev Loading…
1
added flip converter
#822 opened Nov 29, 2022 by jaybdub Loading…
Example link broken in README.md
#806 opened Sep 22, 2022 by charlielito Loading…
QAT : TRT 8 compatible workflow
#804 opened Sep 13, 2022 by SrivastavaKshitij • Draft
1
Fix issues when using interpolate in versions of python >= 3.8
#800 opened Aug 31, 2022 by mahdilamb Loading…
1
[torch2trt/torch2trt.py] Add pretty printing of converters when logging in verbose mode.
#789 opened Aug 13, 2022 by chaoz-dev Loading…
2
[torch.Tensor.__getitem__] Add advanced indexing (GatherND) support to torch.Tensor.__getitem__
#783 opened Aug 4, 2022 by chaoz-dev Loading…
10
added add inplace converter
#774 opened Jul 26, 2022 by jaybdub Loading…
add cpu tracing
#773 opened Jul 26, 2022 by jaybdub Loading…
Enhance the einsum converter
#772 opened Jul 25, 2022 by LongerZrLong Loading…
1
[torch2trt/converters] Add element support to torch.Tensor.__getitem__ converter
#770 opened Jul 23, 2022 by chaoz-dev Loading…
1
4
Fix output table column names and float numbers format
#732 opened May 6, 2022 by spital Loading…
Allow onnx kwargs
#717 opened Apr 6, 2022 by lazToum Loading…
Add torch.Tensor.transpose converter
#708 opened Mar 25, 2022 by kairos03 Loading…
1
expand_as + avg_pool1d converters were added
#699 opened Feb 26, 2022 by Ilyabasharov Loading…
TRTModule Support Dynamic Batch Size with Engine Converted from ONNX
#642 opened Sep 30, 2021 by csgcmai Loading…
decrease memory use by remove unnecessary module inference
#631 opened Sep 20, 2021 by wwqgtxx Loading…
Updated TensorRT Python API in torch2trt
#630 opened Sep 15, 2021 by andi4191 Loading…
Added converter for torch.Tensor.expand_as() method
#614 opened Aug 22, 2021 by maronuu Loading…
2
Added Converters for Basic Casting Operations
#607 opened Aug 11, 2021 by JWLee89 Loading…
4
ProTip! Find all pull requests that aren't related to any open issues with -linked:issue.