search

nod-ai / SHARK-Turbine

Unified compiler/runtime for interfacing with PyTorch Dynamo.
Apache License 2.0
90 stars 44 forks source link

onnx.NonMaxSuppression #650

Closed PhaneeshB closed 1 month ago

PhaneeshB commented 4 months ago

Onnx Doc

Found in this model retinanet_resnet50_fpn_vaiq_int8 https://github.com/nod-ai/SHARK-TestSuite/issues/199

vivekkhandelwal1 commented 4 months ago

This is a vision op hence moving this to a lower priority list as per discussion.

mgehre-amd commented 2 months ago

fyi, this is also appearing at the tail of some Yolo models

AmosLewis commented 2 months ago

PR: https://github.com/llvm/torch-mlir/pull/3501