avg_pool line-71 stride variable is return None. #512
Conversation
This file contains hidden or 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 plugin registry
Update README.md
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
* 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
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.
if you are using avg_pool, stride variable is return None and converting error return you can use stride=kernel_size