added converter for torch.einsum #571
Open
Conversation
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
added support for NC input to batchnorm1d
…tfix list hotfix
removed caffe2 dependency
If ceil_mode is False, the default value of layer.padding_mode is PaddingMode.EXPLICIT_ROUND_DOWN. If ceil_mode is True, padding_mode should be trt.PaddingMode.EXPLICIT_ROUND_UP.
clamp and normalize
consider ceil_mode of torch.nn.MaxPool2d
added mobilenet_v2 to module tests
…actor - adds avg_pool2d - adds max_pool2d - removes AvgPool2d - removes MaxPool2d - adds get_arg(...) - adds torch_dim_to_trt_axes(...) - adds add_trt_constant(...)
* reduced precision docs * support matrix * docs
* Assert that inputs are contiguous * Turn non-contiguous tensors into contiguous * Add unit test * Fix tabs
* Auto-generate custom layer names * fixed layer name count key * updated changelog for adding layer names Co-authored-by: Alex Sergeev <[email protected]>
* added plugin for GroupNorm Co-authored-by: sandeepkumar-skb <[email protected]>
* Remove duplicate filenames which do not work on Windows by merging files * Fix * relu tests Co-authored-by: Koen van de Sande <[email protected]>
…ations (NVIDIA-AI-IOT#505) * Initioal version of ne, floordiv, mod and tensor converters. Extend ops for relu and sigmoid. * Converters for floordiv, mod, ne, and torch::tensor() operations . Extend relu and sigmoid converters to Tensor methods. * Update CHANGELOG.md
…T#482) * added passing of torch2trt_kwargs to conversion context * added passing of torch2trt_kwargs to conversion context
…OT#511) * added filter to floordiv to only enable for pytorch 1.6+ * enabled soft failure for missing torch method
* increment version to 0.2.0 * realse push docs tagfix
* added conv_functional * add Tensor flatten * update changelog for functional conv / flatten * add site to gitignore
… in the file CLA.md of this project. Signed, John Welsh
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
No description provided.