Skip to content
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.

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

added converter for torch.einsum #571

Open
wants to merge 432 commits into
base: master
Choose a base branch
from
This pull request is big! We’re only showing the most recent 250 commits.

Commits on Aug 16, 2019

  1. Merge pull request NVIDIA-AI-IOT#43 from NVIDIA-AI-IOT/batchnorm1d_fix

    added support for NC input to batchnorm1d
    jaybdub authored Aug 16, 2019
    Configuration menu
    Copy the full SHA
    3a02f56 View commit details
    Browse the repository at this point in the history

Commits on Aug 22, 2019

  1. list hotfix

    jaybdub committed Aug 22, 2019
    Configuration menu
    Copy the full SHA
    ecd4893 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    7a960c6 View commit details
    Browse the repository at this point in the history

Commits on Sep 10, 2019

  1. removed caffe2 dependency

    jaybdub committed Sep 10, 2019
    Configuration menu
    Copy the full SHA
    445081f View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#56 from NVIDIA-AI-IOT/pyt1.2

    removed caffe2 dependency
    jaybdub authored Sep 10, 2019
    Configuration menu
    Copy the full SHA
    295961a View commit details
    Browse the repository at this point in the history

Commits on Sep 13, 2019

  1. consider ceil_mode of torch.nn.MaxPool2d

    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.
    mt1871 committed Sep 13, 2019
    Configuration menu
    Copy the full SHA
    e394a14 View commit details
    Browse the repository at this point in the history

Commits on Sep 17, 2019

  1. Configuration menu
    Copy the full SHA
    59e0ec8 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    5ad17bc View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    e951955 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    51e0c95 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    5c41345 View commit details
    Browse the repository at this point in the history
  6. added clamp tensor

    jaybdub committed Sep 17, 2019
    Configuration menu
    Copy the full SHA
    08fecd3 View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2019

  1. added normalize

    jaybdub committed Sep 18, 2019
    Configuration menu
    Copy the full SHA
    2769c29 View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#62 from NVIDIA-AI-IOT/normalize

    clamp and normalize
    jaybdub authored Sep 18, 2019
    Configuration menu
    Copy the full SHA
    493515c View commit details
    Browse the repository at this point in the history
  3. added test cases for MaxPool2d

    mt1871 committed Sep 18, 2019
    Configuration menu
    Copy the full SHA
    43d1b1a View commit details
    Browse the repository at this point in the history

Commits on Sep 19, 2019

  1. Merge pull request NVIDIA-AI-IOT#57 from mt1871/fix_maxpool_ceilmode

    consider ceil_mode of torch.nn.MaxPool2d
    jaybdub authored Sep 19, 2019
    Configuration menu
    Copy the full SHA
    d2825e5 View commit details
    Browse the repository at this point in the history

Commits on Sep 25, 2019

  1. Configuration menu
    Copy the full SHA
    630194c View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#70 from NVIDIA-AI-IOT/mobilenet_v2

    added mobilenet_v2 to module tests
    jaybdub authored Sep 25, 2019
    Configuration menu
    Copy the full SHA
    81e73d2 View commit details
    Browse the repository at this point in the history

Commits on Sep 27, 2019

  1. added parsed_args

    jaybdub committed Sep 27, 2019
    Configuration menu
    Copy the full SHA
    f29ff96 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    d409f13 View commit details
    Browse the repository at this point in the history
  3. dtype to normalized constants

    jaybdub committed Sep 27, 2019
    Configuration menu
    Copy the full SHA
    dc547b8 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    c429ae7 View commit details
    Browse the repository at this point in the history
  5. fixed normalize

    jaybdub committed Sep 27, 2019
    Configuration menu
    Copy the full SHA
    5abcbbe View commit details
    Browse the repository at this point in the history
  6. added missing avg_pool2d.py

    jaybdub committed Sep 27, 2019
    Configuration menu
    Copy the full SHA
    d47af74 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    916e25d View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    100bcdb View commit details
    Browse the repository at this point in the history

Commits on Sep 29, 2019

  1. refactor

    jaybdub committed Sep 29, 2019
    Configuration menu
    Copy the full SHA
    46534bd View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#73 from NVIDIA-AI-IOT/functional_ref…

    …actor
    
    - adds avg_pool2d
    - adds max_pool2d
    - removes AvgPool2d
    - removes MaxPool2d
    - adds get_arg(...)
    - adds torch_dim_to_trt_axes(...)
    - adds add_trt_constant(...)
    jaybdub authored Sep 29, 2019
    Configuration menu
    Copy the full SHA
    83f1765 View commit details
    Browse the repository at this point in the history

Commits on Sep 30, 2019

  1. added split and chunk

    jaybdub committed Sep 30, 2019
    Configuration menu
    Copy the full SHA
    05e7022 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    d6a3fd5 View commit details
    Browse the repository at this point in the history
  3. Merge pull request NVIDIA-AI-IOT#74 from NVIDIA-AI-IOT/split

    - adds ``torch.chunk`` and ``torch.Tensor.chunk``
    - adds ``torch.split`` and ``torch.Tensor.split``
    - adds tests for ``squeezenet*`` models
    jaybdub authored Sep 30, 2019
    Configuration menu
    Copy the full SHA
    9b616d7 View commit details
    Browse the repository at this point in the history
  4. added permute

    jaybdub committed Sep 30, 2019
    Configuration menu
    Copy the full SHA
    a838a19 View commit details
    Browse the repository at this point in the history
  5. fixed permute converter name

    jaybdub committed Sep 30, 2019
    Configuration menu
    Copy the full SHA
    7b46e02 View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    c0c89d2 View commit details
    Browse the repository at this point in the history

Commits on Oct 1, 2019

  1. Configuration menu
    Copy the full SHA
    72f0cd7 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    0bfe27d View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    5699dcd View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    4c21936 View commit details
    Browse the repository at this point in the history
  5. Configuration menu
    Copy the full SHA
    9745e31 View commit details
    Browse the repository at this point in the history
  6. added rdiv and radd

    jaybdub committed Oct 1, 2019
    Configuration menu
    Copy the full SHA
    9bcadbb View commit details
    Browse the repository at this point in the history
  7. added rdiv, rmul, radd

    jaybdub committed Oct 1, 2019
    Configuration menu
    Copy the full SHA
    3d2fef8 View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    e69c577 View commit details
    Browse the repository at this point in the history
  9. Merge pull request NVIDIA-AI-IOT#80 from NVIDIA-AI-IOT/prepare_broadc…

    …astable_tensors
    
    Prepare broadcastable tensors
    jaybdub authored Oct 1, 2019
    Configuration menu
    Copy the full SHA
    fec6b22 View commit details
    Browse the repository at this point in the history
  10. added mnasnet to test

    jaybdub committed Oct 1, 2019
    Configuration menu
    Copy the full SHA
    f27aa27 View commit details
    Browse the repository at this point in the history
  11. Merge pull request NVIDIA-AI-IOT#81 from NVIDIA-AI-IOT/prepare_broadc…

    …astable_tensors
    
    added mnasnet to test
    jaybdub authored Oct 1, 2019
    Configuration menu
    Copy the full SHA
    922a0f7 View commit details
    Browse the repository at this point in the history

Commits on Oct 2, 2019

  1. refactored tests

    jaybdub committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    f29e1c9 View commit details
    Browse the repository at this point in the history
  2. refactored tests

    jaybdub committed Oct 2, 2019
    Configuration menu
    Copy the full SHA
    5b1d552 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    a660786 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    4852b82 View commit details
    Browse the repository at this point in the history
  5. Merge pull request NVIDIA-AI-IOT#84 from NVIDIA-AI-IOT/extra_tests

    refactored tests
    
    - adds --include flag to test.py to load additional tests
    - moves torchivsion tests from torch2trt.module_test to torch2trt.tests.torchvision.classification
    - adds torchvision segmentation tests to torch2trt.tests.torchvision.segmentation
    jaybdub authored Oct 2, 2019
    Configuration menu
    Copy the full SHA
    f6eae5a View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    7693e81 View commit details
    Browse the repository at this point in the history
  7. Configuration menu
    Copy the full SHA
    f2c931e View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    bbe7b1d View commit details
    Browse the repository at this point in the history
  9. Configuration menu
    Copy the full SHA
    506fb87 View commit details
    Browse the repository at this point in the history
  10. Configuration menu
    Copy the full SHA
    beccc1a View commit details
    Browse the repository at this point in the history
  11. Merge pull request NVIDIA-AI-IOT#10 from narendasan/narendasan/typo_i…

    …n_readme
    
    Missing an import in the first example
    jaybdub authored Oct 2, 2019
    Configuration menu
    Copy the full SHA
    c7cb4bd View commit details
    Browse the repository at this point in the history
  12. Configuration menu
    Copy the full SHA
    c982944 View commit details
    Browse the repository at this point in the history
  13. Configuration menu
    Copy the full SHA
    887ca21 View commit details
    Browse the repository at this point in the history
  14. removes caffe2 from build

    caffe2 is not needed and breaks build when using PyTorch 1.2
    jaybdub authored Oct 2, 2019
    Configuration menu
    Copy the full SHA
    6fafa46 View commit details
    Browse the repository at this point in the history
  15. Merge pull request NVIDIA-AI-IOT#12 from narendasan/narendasan/custom…

    …_dep_locations
    
    Adds a way to specify custom library locations and remove build artifacts
    jaybdub authored Oct 2, 2019
    Configuration menu
    Copy the full SHA
    950109b View commit details
    Browse the repository at this point in the history

Commits on Oct 4, 2019

  1. added support for prelu

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    c8b3619 View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#88 from NVIDIA-AI-IOT/prelu

    added support for prelu
    jaybdub authored Oct 4, 2019
    Configuration menu
    Copy the full SHA
    b962b4d View commit details
    Browse the repository at this point in the history
  3. added supported unary ops

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    9bbf348 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#89 from NVIDIA-AI-IOT/unary_supported

    added supported unary ops
    jaybdub authored Oct 4, 2019
    Configuration menu
    Copy the full SHA
    972a7c7 View commit details
    Browse the repository at this point in the history
  5. added max reduce/elementwise

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    1c87da6 View commit details
    Browse the repository at this point in the history
  6. added min reduce/elementwise

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    86feae5 View commit details
    Browse the repository at this point in the history
  7. added sub

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    dfc7483 View commit details
    Browse the repository at this point in the history
  8. added pow elementwise

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    601b94e View commit details
    Browse the repository at this point in the history
  9. fixed min/max reduce all

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    7947855 View commit details
    Browse the repository at this point in the history
  10. added prod

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    0a3eaee View commit details
    Browse the repository at this point in the history
  11. added sum reduce

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    16ac488 View commit details
    Browse the repository at this point in the history
  12. increment patch

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    95a5849 View commit details
    Browse the repository at this point in the history
  13. Configuration menu
    Copy the full SHA
    8d00907 View commit details
    Browse the repository at this point in the history
  14. more relu interfaces

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    a6ee9c4 View commit details
    Browse the repository at this point in the history
  15. Merge pull request NVIDIA-AI-IOT#90 from NVIDIA-AI-IOT/supported_elem…

    …entwise
    
    Supported elementwise
    jaybdub authored Oct 4, 2019
    Configuration menu
    Copy the full SHA
    28e1700 View commit details
    Browse the repository at this point in the history
  16. Configuration menu
    Copy the full SHA
    599524b View commit details
    Browse the repository at this point in the history
  17. Configuration menu
    Copy the full SHA
    f25eaa5 View commit details
    Browse the repository at this point in the history
  18. removed leaf check

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    59ee834 View commit details
    Browse the repository at this point in the history
  19. Configuration menu
    Copy the full SHA
    d8b43bd View commit details
    Browse the repository at this point in the history
  20. Configuration menu
    Copy the full SHA
    d7d967c View commit details
    Browse the repository at this point in the history
  21. Configuration menu
    Copy the full SHA
    9e8753a View commit details
    Browse the repository at this point in the history
  22. added save_load test

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    713086d View commit details
    Browse the repository at this point in the history
  23. added save/load test

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    2d373fe View commit details
    Browse the repository at this point in the history
  24. Merge pull request NVIDIA-AI-IOT#93 from NVIDIA-AI-IOT/py2_serializat…

    …ion_fix
    
    Py2 serialization fix
    jaybdub authored Oct 4, 2019
    Configuration menu
    Copy the full SHA
    06ccc64 View commit details
    Browse the repository at this point in the history
  25. extra ignore

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    4b5a3c2 View commit details
    Browse the repository at this point in the history
  26. Configuration menu
    Copy the full SHA
    82e694c View commit details
    Browse the repository at this point in the history
  27. added leaky_relu

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    6c403fc View commit details
    Browse the repository at this point in the history
  28. added elu

    jaybdub committed Oct 4, 2019
    Configuration menu
    Copy the full SHA
    11c465c View commit details
    Browse the repository at this point in the history

Commits on Oct 5, 2019

  1. added selu, softsign, elu

    jaybdub committed Oct 5, 2019
    Configuration menu
    Copy the full SHA
    e2ebb8e View commit details
    Browse the repository at this point in the history
  2. added softplus

    jaybdub committed Oct 5, 2019
    Configuration menu
    Copy the full SHA
    9b3a3c4 View commit details
    Browse the repository at this point in the history
  3. softplus, softsign

    jaybdub committed Oct 5, 2019
    Configuration menu
    Copy the full SHA
    2cb97a3 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#95 from NVIDIA-AI-IOT/activations_su…

    …pported
    
    Activations supported
    jaybdub authored Oct 5, 2019
    Configuration menu
    Copy the full SHA
    4046625 View commit details
    Browse the repository at this point in the history

Commits on Oct 8, 2019

  1. Configuration menu
    Copy the full SHA
    384384e View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    e0b0e71 View commit details
    Browse the repository at this point in the history
  3. merged

    jaybdub committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    345b561 View commit details
    Browse the repository at this point in the history
  4. graphviz module prefix fix

    jaybdub committed Oct 8, 2019
    Configuration menu
    Copy the full SHA
    323f1d3 View commit details
    Browse the repository at this point in the history

Commits on Oct 20, 2019

  1. fixed Linear w.o bias

    jaybdub committed Oct 20, 2019
    Configuration menu
    Copy the full SHA
    e83730e View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#125 from jaybdub/linear_no_bias

    fixed Linear w.o bias
    jaybdub authored Oct 20, 2019
    Configuration menu
    Copy the full SHA
    7255a24 View commit details
    Browse the repository at this point in the history

Commits on Oct 21, 2019

  1. Configuration menu
    Copy the full SHA
    3fac66b View commit details
    Browse the repository at this point in the history
  2. added profiling

    jaybdub committed Oct 21, 2019
    Configuration menu
    Copy the full SHA
    c76f141 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    ace0abd View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    f7dbb96 View commit details
    Browse the repository at this point in the history
  5. graphviz module prefix fix

    jaybdub committed Oct 21, 2019
    Configuration menu
    Copy the full SHA
    84101a9 View commit details
    Browse the repository at this point in the history
  6. Configuration menu
    Copy the full SHA
    97260f2 View commit details
    Browse the repository at this point in the history
  7. added profiling

    jaybdub committed Oct 21, 2019
    Configuration menu
    Copy the full SHA
    0a2c3eb View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    dc620d9 View commit details
    Browse the repository at this point in the history
  9. Merge pull request NVIDIA-AI-IOT#106 from NVIDIA-AI-IOT/network_visua…

    …lization
    
    Network visualization
    jaybdub authored Oct 21, 2019
    Configuration menu
    Copy the full SHA
    7984972 View commit details
    Browse the repository at this point in the history

Commits on Oct 28, 2019

  1. added dummy

    jaybdub committed Oct 28, 2019
    Configuration menu
    Copy the full SHA
    e520dd2 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    5ab1561 View commit details
    Browse the repository at this point in the history
  3. incremented patch

    jaybdub committed Oct 28, 2019
    Configuration menu
    Copy the full SHA
    b967c99 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#138 from NVIDIA-AI-IOT/unsupported_m…

    …ethods
    
    Unsupported methods
    jaybdub authored Oct 28, 2019
    Configuration menu
    Copy the full SHA
    53ce056 View commit details
    Browse the repository at this point in the history

Commits on Nov 19, 2019

  1. fixed upsampling<->interpolate translation

    Kshitij Srivastava committed Nov 19, 2019
    Configuration menu
    Copy the full SHA
    a32e684 View commit details
    Browse the repository at this point in the history
  2. reverted changes to fix build

    Kshitij Srivastava committed Nov 19, 2019
    Configuration menu
    Copy the full SHA
    af56018 View commit details
    Browse the repository at this point in the history

Commits on Nov 23, 2019

  1. Configuration menu
    Copy the full SHA
    4da14ed View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    33af10d View commit details
    Browse the repository at this point in the history

Commits on Nov 26, 2019

  1. added int8 support

    jaybdub committed Nov 26, 2019
    Configuration menu
    Copy the full SHA
    112922b View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    46bb0df View commit details
    Browse the repository at this point in the history
  3. handle loose dataset types

    jaybdub committed Nov 26, 2019
    Configuration menu
    Copy the full SHA
    a7c35d5 View commit details
    Browse the repository at this point in the history

Commits on Nov 27, 2019

  1. Configuration menu
    Copy the full SHA
    8e6e475 View commit details
    Browse the repository at this point in the history
  2. fix int8_support python2

    jaybdub committed Nov 27, 2019
    Configuration menu
    Copy the full SHA
    b8a54ac View commit details
    Browse the repository at this point in the history

Commits on Dec 3, 2019

  1. Configuration menu
    Copy the full SHA
    f60509b View commit details
    Browse the repository at this point in the history
  2. increment vers

    jaybdub committed Dec 3, 2019
    Configuration menu
    Copy the full SHA
    3a05d4a View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    eb54b5f View commit details
    Browse the repository at this point in the history

Commits on Dec 12, 2019

  1. Configuration menu
    Copy the full SHA
    23ac868 View commit details
    Browse the repository at this point in the history

Commits on Dec 13, 2019

  1. instance norm

    jaybdub committed Dec 13, 2019
    Configuration menu
    Copy the full SHA
    5f30709 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    67ab729 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    b3e0c08 View commit details
    Browse the repository at this point in the history

Commits on Dec 14, 2019

  1. Configuration menu
    Copy the full SHA
    c374807 View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#195 from NVIDIA-AI-IOT/getitem

    added converter / tests for torch.Tensor.__getitem__
    jaybdub authored Dec 14, 2019
    Configuration menu
    Copy the full SHA
    1ee2c05 View commit details
    Browse the repository at this point in the history
  3. fixed Linear for N, *, H case

    jaybdub committed Dec 14, 2019
    Configuration menu
    Copy the full SHA
    1a26b54 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#196 from NVIDIA-AI-IOT/linear_fix

    fixed Linear for N, *, H case
    jaybdub authored Dec 14, 2019
    Configuration menu
    Copy the full SHA
    190a84d View commit details
    Browse the repository at this point in the history

Commits on Dec 17, 2019

  1. fixed permute for list args

    jaybdub committed Dec 17, 2019
    Configuration menu
    Copy the full SHA
    8142c6f View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#206 from NVIDIA-AI-IOT/permute_lista…

    …rg_fix
    
    fixed permute for list args
    jaybdub authored Dec 17, 2019
    Configuration menu
    Copy the full SHA
    07831d6 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    72f6ec2 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#207 from NVIDIA-AI-IOT/adaptive_max_…

    …pool2d
    
    added support for adaptive_max_pool2d using regular pooling
    jaybdub authored Dec 17, 2019
    Configuration menu
    Copy the full SHA
    d526b24 View commit details
    Browse the repository at this point in the history

Commits on Jan 6, 2020

  1. update clamp

    SnowMasaya committed Jan 6, 2020
    Configuration menu
    Copy the full SHA
    c6d047f View commit details
    Browse the repository at this point in the history
  2. rm idea

    SnowMasaya committed Jan 6, 2020
    Configuration menu
    Copy the full SHA
    1756aa9 View commit details
    Browse the repository at this point in the history

Commits on Jan 7, 2020

  1. Merge pull request NVIDIA-AI-IOT#224 from SnowMasaya/clamp

    Clamp isolated min/max
    jaybdub authored Jan 7, 2020
    Configuration menu
    Copy the full SHA
    e22844a View commit details
    Browse the repository at this point in the history

Commits on Apr 10, 2020

  1. plugin setup.py

    jaybdub committed Apr 10, 2020
    Configuration menu
    Copy the full SHA
    6be6da4 View commit details
    Browse the repository at this point in the history

Commits on Apr 28, 2020

  1. asdf

    jaybdub committed Apr 28, 2020
    Configuration menu
    Copy the full SHA
    1142079 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    b5bb91d View commit details
    Browse the repository at this point in the history

Commits on Apr 29, 2020

  1. version handling

    jaybdub committed Apr 29, 2020
    Configuration menu
    Copy the full SHA
    62a8918 View commit details
    Browse the repository at this point in the history

Commits on May 20, 2020

  1. Configuration menu
    Copy the full SHA
    1ca350b View commit details
    Browse the repository at this point in the history

Commits on May 26, 2020

  1. added upsample code in converters

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    fb72cac View commit details
    Browse the repository at this point in the history
  2. added functional <--> nn.module binding for upsample

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    f6a7eaf View commit details
    Browse the repository at this point in the history
  3. made changes:

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    3ecd4f5 View commit details
    Browse the repository at this point in the history
  4. working upsample with trt6

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    3c00f09 View commit details
    Browse the repository at this point in the history
  5. added F.interpolate converter

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    558cbff View commit details
    Browse the repository at this point in the history
  6. (WIP), trt version check

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    f48ad4b View commit details
    Browse the repository at this point in the history
  7. Add 3d convolutions and interpolations

    oliver-batchelor authored and Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    8f63b22 View commit details
    Browse the repository at this point in the history
  8. Update lots of ops

    oliver-batchelor authored and Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    3ebf1b9 View commit details
    Browse the repository at this point in the history
  9. More opreations

    oliver-batchelor authored and Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    63e65d4 View commit details
    Browse the repository at this point in the history
  10. Remove tools import

    oliver-batchelor authored and Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    a4f0b52 View commit details
    Browse the repository at this point in the history
  11. fixed trt version compatibility for torch2trt.py

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    393a13d View commit details
    Browse the repository at this point in the history
  12. separated trt 7 ops

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    401be2e View commit details
    Browse the repository at this point in the history
  13. added original ops

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    8cb3606 View commit details
    Browse the repository at this point in the history
  14. reverting to default for trt 7

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    3ce1371 View commit details
    Browse the repository at this point in the history
  15. changed value from 7 to 7.0

    Kshitij Srivastava committed May 26, 2020
    Configuration menu
    Copy the full SHA
    af2674c View commit details
    Browse the repository at this point in the history

Commits on Jun 4, 2020

  1. added unit test for ConvTranspose2d and fixed an indentation error in…

    … parent PR
    Kshitij Srivastava committed Jun 4, 2020
    Configuration menu
    Copy the full SHA
    3c16ea5 View commit details
    Browse the repository at this point in the history
  2. added testcase for commonly used kernel size 4, stride 2, padding 1 c…

    …onvtranspose configuration
    jaybdub committed Jun 4, 2020
    Configuration menu
    Copy the full SHA
    512a511 View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    a1daf58 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    f3881cc View commit details
    Browse the repository at this point in the history

Commits on Jun 9, 2020

  1. Configuration menu
    Copy the full SHA
    b2c8c13 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    5c31892 View commit details
    Browse the repository at this point in the history
  3. merged interpolate modules

    jaybdub committed Jun 9, 2020
    Configuration menu
    Copy the full SHA
    e3f45d3 View commit details
    Browse the repository at this point in the history
  4. format init

    jaybdub committed Jun 9, 2020
    Configuration menu
    Copy the full SHA
    1f4f69c View commit details
    Browse the repository at this point in the history
  5. increment minor vers

    jaybdub committed Jun 9, 2020
    Configuration menu
    Copy the full SHA
    229f807 View commit details
    Browse the repository at this point in the history

Commits on Jun 10, 2020

  1. added --plugins to setup

    jaybdub committed Jun 10, 2020
    Configuration menu
    Copy the full SHA
    57c8188 View commit details
    Browse the repository at this point in the history
  2. plugins disclaimer to readme

    jaybdub committed Jun 10, 2020
    Configuration menu
    Copy the full SHA
    a22e82c View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    1a5a397 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    b31611c View commit details
    Browse the repository at this point in the history
  5. Merge pull request NVIDIA-AI-IOT#332 from NVIDIA-AI-IOT/SrivastavaKsh…

    …itij-new_trt_ops
    
    Srivastava kshitij new trt ops
    jaybdub authored Jun 10, 2020
    Configuration menu
    Copy the full SHA
    1f66266 View commit details
    Browse the repository at this point in the history

Commits on Jun 11, 2020

  1. Configuration menu
    Copy the full SHA
    75fd8e2 View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#337 from NVIDIA-AI-IOT/plugin_silent…

    …fail
    
    allowed plugin loading to silent fail
    jaybdub authored Jun 11, 2020
    Configuration menu
    Copy the full SHA
    bea0f6a View commit details
    Browse the repository at this point in the history
  3. Configuration menu
    Copy the full SHA
    22e1118 View commit details
    Browse the repository at this point in the history
  4. Merge pull request NVIDIA-AI-IOT#338 from NVIDIA-AI-IOT/onnx_alternative

    added ONNX conversion alternative to torch2trt
    jaybdub authored Jun 11, 2020
    Configuration menu
    Copy the full SHA
    fe76a7b View commit details
    Browse the repository at this point in the history

Commits on Jul 2, 2020

  1. Add requirements.txt

    vellamike authored Jul 2, 2020
    Configuration menu
    Copy the full SHA
    b0a7724 View commit details
    Browse the repository at this point in the history
  2. Added install information

    Added information regarding installing dependencies
    vellamike authored Jul 2, 2020
    Configuration menu
    Copy the full SHA
    2aa01dd View commit details
    Browse the repository at this point in the history

Commits on Jul 6, 2020

  1. Update README.md

    FIxed error in instrucitons
    vellamike authored Jul 6, 2020
    Configuration menu
    Copy the full SHA
    ed37a06 View commit details
    Browse the repository at this point in the history

Commits on Jul 8, 2020

  1. readme tensorrt install tip

    jaybdub committed Jul 8, 2020
    Configuration menu
    Copy the full SHA
    f7ff467 View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#361 from NVIDIA-AI-IOT/vellamike-upd…

    …ate-requirements
    
    Vellamike update requirements
    jaybdub authored Jul 8, 2020
    Configuration menu
    Copy the full SHA
    d88fcd5 View commit details
    Browse the repository at this point in the history

Commits on Aug 25, 2020

  1. added preliminary code for narrow

    Kshitij Srivastava committed Aug 25, 2020
    Configuration menu
    Copy the full SHA
    f9828cc View commit details
    Browse the repository at this point in the history

Commits on Aug 28, 2020

  1. narrow converter added successfully

    Kshitij Srivastava committed Aug 28, 2020
    Configuration menu
    Copy the full SHA
    5c4ad8f View commit details
    Browse the repository at this point in the history

Commits on Aug 31, 2020

  1. softmax neg dim

    jaybdub committed Aug 31, 2020
    Configuration menu
    Copy the full SHA
    f3fe46c View commit details
    Browse the repository at this point in the history
  2. Merge pull request NVIDIA-AI-IOT#396 from NVIDIA-AI-IOT/softmax_negdi…

    …m_fix
    
    softmax neg dim
    jaybdub authored Aug 31, 2020
    Configuration menu
    Copy the full SHA
    1f1701d View commit details
    Browse the repository at this point in the history
  3. Merge pull request NVIDIA-AI-IOT#393 from SrivastavaKshitij/narrow_co…

    …nverter
    
    Narrow converter
    jaybdub authored Aug 31, 2020
    Configuration menu
    Copy the full SHA
    63895f0 View commit details
    Browse the repository at this point in the history

Commits on Sep 1, 2020

  1. Configuration menu
    Copy the full SHA
    d245496 View commit details
    Browse the repository at this point in the history
  2. bugfix and add

    jaybdub committed Sep 1, 2020
    Configuration menu
    Copy the full SHA
    fb8da9a View commit details
    Browse the repository at this point in the history
  3. add tests

    jaybdub committed Sep 1, 2020
    Configuration menu
    Copy the full SHA
    d6e6f3b View commit details
    Browse the repository at this point in the history
  4. div

    jaybdub committed Sep 1, 2020
    Configuration menu
    Copy the full SHA
    d14c6b7 View commit details
    Browse the repository at this point in the history
  5. dimfix

    jaybdub committed Sep 1, 2020
    Configuration menu
    Copy the full SHA
    a08e6dc View commit details
    Browse the repository at this point in the history

Commits on Sep 2, 2020

  1. dump converters script

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    742b3fd View commit details
    Browse the repository at this point in the history
  2. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    765c3ad View commit details
    Browse the repository at this point in the history
  3. changelog contributing

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    3adbbe1 View commit details
    Browse the repository at this point in the history
  4. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    3beeae2 View commit details
    Browse the repository at this point in the history
  5. typos

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    422e1e6 View commit details
    Browse the repository at this point in the history
  6. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    4933faf View commit details
    Browse the repository at this point in the history
  7. expose the batch size of INT8 calibration as parameter, since different

    size may generate different accuracy loss.
    Chujingjun committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    3ddd70d View commit details
    Browse the repository at this point in the history
  8. Configuration menu
    Copy the full SHA
    48eb79e View commit details
    Browse the repository at this point in the history
  9. Merge pull request NVIDIA-AI-IOT#398 from Chujingjun/master

    expose the batch size of INT8 calibration as parameter
    jaybdub authored Sep 2, 2020
    Configuration menu
    Copy the full SHA
    f5fb752 View commit details
    Browse the repository at this point in the history
  10. dump converters script

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    16492db View commit details
    Browse the repository at this point in the history
  11. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    88416f9 View commit details
    Browse the repository at this point in the history
  12. changelog contributing

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    1a59749 View commit details
    Browse the repository at this point in the history
  13. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    93abd9f View commit details
    Browse the repository at this point in the history
  14. typos

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    384ac89 View commit details
    Browse the repository at this point in the history
  15. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    656805b View commit details
    Browse the repository at this point in the history
  16. docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    571cb00 View commit details
    Browse the repository at this point in the history
  17. docs build contrib

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    a263fe0 View commit details
    Browse the repository at this point in the history
  18. build docs

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    0655a80 View commit details
    Browse the repository at this point in the history
  19. asdf

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    1474eee View commit details
    Browse the repository at this point in the history
  20. docs merge

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    3d496ae View commit details
    Browse the repository at this point in the history
  21. merge fixes

    jaybdub committed Sep 2, 2020
    Configuration menu
    Copy the full SHA
    88d8038 View commit details
    Browse the repository at this point in the history

Commits on Sep 3, 2020

  1. asdf

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    5160901 View commit details
    Browse the repository at this point in the history
  2. mkdocs

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    5aae576 View commit details
    Browse the repository at this point in the history
  3. docs

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    86d9c3a View commit details
    Browse the repository at this point in the history
  4. contrib plain md

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    67de214 View commit details
    Browse the repository at this point in the history
  5. basic usage

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    6c0aec7 View commit details
    Browse the repository at this point in the history
  6. issues pathfix

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    bd2cd11 View commit details
    Browse the repository at this point in the history
  7. broken link

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    50169ef View commit details
    Browse the repository at this point in the history
  8. release build docs

    jaybdub committed Sep 3, 2020
    Configuration menu
    Copy the full SHA
    62afaf1 View commit details
    Browse the repository at this point in the history
  9. Configuration menu
    Copy the full SHA
    c9dcd5e View commit details
    Browse the repository at this point in the history
  10. Reduce precision docs (NVIDIA-AI-IOT#400)

    * reduced precision docs
    
    * support matrix
    
    * docs
    jaybdub authored Sep 3, 2020
    Configuration menu
    Copy the full SHA
    3efb89d View commit details
    Browse the repository at this point in the history
  11. Configuration menu
    Copy the full SHA
    4141730 View commit details
    Browse the repository at this point in the history
  12. Update mkdocs.yml

    jaybdub authored Sep 3, 2020
    Configuration menu
    Copy the full SHA
    7563770 View commit details
    Browse the repository at this point in the history

Commits on Sep 15, 2020

  1. Configuration menu
    Copy the full SHA
    ea835dc View commit details
    Browse the repository at this point in the history

Commits on Sep 18, 2020

  1. Assert that inputs are contiguous (NVIDIA-AI-IOT#418)

    * Assert that inputs are contiguous
    
    * Turn non-contiguous tensors into contiguous
    
    * Add unit test
    
    * Fix tabs
    alsrgv authored Sep 18, 2020
    Configuration menu
    Copy the full SHA
    b0cc8e7 View commit details
    Browse the repository at this point in the history

Commits on Oct 19, 2020

  1. Add layer names (NVIDIA-AI-IOT#432)

    * Auto-generate custom layer names
    
    * fixed layer name count key
    
    * updated changelog for adding layer names
    
    Co-authored-by: Alex Sergeev <[email protected]>
    jaybdub and alsrgv authored Oct 19, 2020
    Configuration menu
    Copy the full SHA
    a9a6a53 View commit details
    Browse the repository at this point in the history

Commits on Nov 4, 2020

  1. Sandeepkumar skb groupnorm plugin (NVIDIA-AI-IOT#437)

    * added plugin for GroupNorm
    
    Co-authored-by: sandeepkumar-skb <[email protected]>
    jaybdub and sandeepkumar-skb authored Nov 4, 2020
    Configuration menu
    Copy the full SHA
    d1fa6f9 View commit details
    Browse the repository at this point in the history

Commits on Nov 17, 2020

  1. Koenvandesande remove duplicate filenames (NVIDIA-AI-IOT#448)

    * Remove duplicate filenames which do not work on Windows by merging files
    
    * Fix
    
    * relu tests
    
    Co-authored-by: Koen van de Sande <[email protected]>
    jaybdub and koenvandesande authored Nov 17, 2020
    Configuration menu
    Copy the full SHA
    adccbf1 View commit details
    Browse the repository at this point in the history

Commits on Dec 2, 2020

  1. Configuration menu
    Copy the full SHA
    81024cc View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    2b1827e View commit details
    Browse the repository at this point in the history

Commits on Jan 13, 2021

  1. Configuration menu
    Copy the full SHA
    033df0c View commit details
    Browse the repository at this point in the history

Commits on Feb 22, 2021

  1. Additional converters for floordiv, mod, ne, and torch::tensor() oper…

    …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
    meremeev authored Feb 22, 2021
    Configuration menu
    Copy the full SHA
    3e4ed64 View commit details
    Browse the repository at this point in the history

Commits on Feb 25, 2021

  1. added passing of torch2trt_kwargs to conversion context (NVIDIA-AI-IO…

    …T#482)
    
    * added passing of torch2trt_kwargs to conversion context
    
    * added passing of torch2trt_kwargs to conversion context
    jaybdub authored Feb 25, 2021
    Configuration menu
    Copy the full SHA
    a5bdd29 View commit details
    Browse the repository at this point in the history
  2. added filter to floordiv to only enable for pytorch 1.6+ (NVIDIA-AI-I…

    …OT#511)
    
    * added filter to floordiv to only enable for pytorch 1.6+
    
    * enabled soft failure for missing torch method
    jaybdub authored Feb 25, 2021
    Configuration menu
    Copy the full SHA
    e55a7a5 View commit details
    Browse the repository at this point in the history

Commits on Mar 2, 2021

  1. Configuration menu
    Copy the full SHA
    15da623 View commit details
    Browse the repository at this point in the history
  2. Release push docs tagfix (NVIDIA-AI-IOT#519)

    * increment version to 0.2.0
    
    * realse push docs tagfix
    jaybdub authored Mar 2, 2021
    Configuration menu
    Copy the full SHA
    44977a9 View commit details
    Browse the repository at this point in the history

Commits on Mar 20, 2021

  1. Update README.md

    jaybdub authored Mar 20, 2021
    Configuration menu
    Copy the full SHA
    8100c6a View commit details
    Browse the repository at this point in the history
  2. Create CLA.md

    jaybdub authored Mar 20, 2021
    Configuration menu
    Copy the full SHA
    72c81f9 View commit details
    Browse the repository at this point in the history

Commits on Mar 25, 2021

  1. Functional conv (NVIDIA-AI-IOT#535)

    * added conv_functional
    
    * add Tensor flatten
    
    * update changelog for functional conv / flatten
    
    * add site to gitignore
    jaybdub authored Mar 25, 2021
    Configuration menu
    Copy the full SHA
    b9d9e73 View commit details
    Browse the repository at this point in the history

Commits on Apr 26, 2021

  1. added contributors page

    jaybdub committed Apr 26, 2021
    Configuration menu
    Copy the full SHA
    8f7f4f7 View commit details
    Browse the repository at this point in the history
  2. Configuration menu
    Copy the full SHA
    2d6150a View commit details
    Browse the repository at this point in the history
  3. I have read and agree to the Contributor License Agreement as written…

    … in the file CLA.md of this project. Signed, John Welsh
    jaybdub committed Apr 26, 2021
    Configuration menu
    Copy the full SHA
    1dc9ad1 View commit details
    Browse the repository at this point in the history
  4. Configuration menu
    Copy the full SHA
    e01279c View commit details
    Browse the repository at this point in the history

Commits on Jun 15, 2021

  1. added silu converter

    John Welsh committed Jun 15, 2021
    Configuration menu
    Copy the full SHA
    2653a71 View commit details
    Browse the repository at this point in the history
  2. functional linear converter

    John Welsh committed Jun 15, 2021
    Configuration menu
    Copy the full SHA
    fb67b42 View commit details
    Browse the repository at this point in the history
  3. added gelu converter

    John Welsh committed Jun 15, 2021
    Configuration menu
    Copy the full SHA
    5ee6033 View commit details
    Browse the repository at this point in the history
  4. added layer_norm converter

    John Welsh committed Jun 15, 2021
    Configuration menu
    Copy the full SHA
    ddb3558 View commit details
    Browse the repository at this point in the history

Commits on Jun 16, 2021

  1. added converter for torch.einsum

    John Welsh committed Jun 16, 2021
    Configuration menu
    Copy the full SHA
    8ce974c View commit details
    Browse the repository at this point in the history
  2. einsum

    John Welsh committed Jun 16, 2021
    Configuration menu
    Copy the full SHA
    2e4ccd3 View commit details
    Browse the repository at this point in the history