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.

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

Support NMS op lowering #3871

Open
wants to merge 10 commits into
base: main
Choose a base branch
from
Open

Support NMS op lowering #3871

wants to merge 10 commits into from

Commits on Nov 21, 2024

  1. Add TorchToLinalg lowering of torchvision.nms op

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    0e3adf3 View commit details
    Browse the repository at this point in the history

  2. Switch to torch decomposition from linalg lowering

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    72fa39b View commit details
    Browse the repository at this point in the history

  3. Add workaround of masked_select op and fix a bug of updating mask

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    9752b92 View commit details
    Browse the repository at this point in the history

  4. Fix NMS op OnnxToTorch lowering

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    0b73a8d View commit details
    Browse the repository at this point in the history

  5. Fix format

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    9a5072c View commit details
    Browse the repository at this point in the history

  6. Fix insert index when updating result

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    c660775 View commit details
    Browse the repository at this point in the history

  7. Fix onnx.nms lit tests

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    e6efb51 View commit details
    Browse the repository at this point in the history

  8. Optimize IoU calculation

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    822d929 View commit details
    Browse the repository at this point in the history

  9. Fix lit tests

    @jinchen62
    jinchen62 committed Nov 21, 2024
    Configuration menu
    Copy the full SHA
    23869b4 View commit details
    Browse the repository at this point in the history

Commits on Nov 22, 2024

  1. Fix dynamic shape issue

    @jinchen62
    jinchen62 committed Nov 22, 2024
    Configuration menu
    Copy the full SHA
    bbbfe87 View commit details
    Browse the repository at this point in the history