Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • ckirchhoff's avatar
    [Feature] Add support for Ascend devices with nms_rotated (#2550) · 54ed0ed8
    ckirchhoff authored 
    * [Feature]: add nms_rotated npu adaptater code

* [BugFix]: modify param in nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated.cpp

* [Doc]: add nms_rotated op in supported op list at ops.md

* [Test]: add nms_rotated unit_test

* [Bug]: remove device parameter in test_batched_nms function
    54ed0ed8
test_nms_rotated.py 5.12 KB