Skip to content

↔ [Converter] Add support for torch.ops.aten._native_batch_norm_legit_no_training.default in Torch-TensorRT #2444

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Closed
gs-olive opened this issue Nov 7, 2023 · 1 comment · Fixed by #2446
Closed

↔ [Converter] Add support for torch.ops.aten._native_batch_norm_legit_no_training.default in Torch-TensorRT #2444

gs-olive opened this issue Nov 7, 2023 · 1 comment · Fixed by #2446
Assignees
@zewenli98
Labels
component: converters Issues re: Specific op converters feature request New feature or request

Comments

@gs-olive
Copy link
Collaborator

gs-olive commented Nov 7, 2023

  • Newly appearing in resnet models in latest Torch nightly versions
  • Likely can be stacked with existing converter
@gs-olive gs-olive added feature request New feature or request component: converters Issues re: Specific op converters labels Nov 7, 2023
@zewenli98
Copy link
Collaborator

PR #2446

@gs-olive gs-olive linked a pull request Nov 8, 2023 that will close this issue
chore: add additional BN native converter #2446
Merged
7 tasks
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@zewenli98 zewenli98

Labels
component: converters Issues re: Specific op converters feature request New feature or request
Projects
None yet
Milestone
No milestone
Development

Successfully merging a pull request may close this issue.

chore: add additional BN native converter pytorch/TensorRT
2 participants
@zewenli98 @gs-olive