Refactor and implement new nms.

pytorch · Mar 23, 2024 · 302cd4c · 302cd4c
1 parent e237888
commit 302cd4c
Show file tree

Hide file tree

Showing 10 changed files with 308 additions and 376 deletions.
diff --git a/torch_xla/csrc/BUILD b/torch_xla/csrc/BUILD
@@ -66,6 +66,7 @@ ptxla_cc_library(
         "xla_lower_util.cpp",
         "xla_op_builder.cpp",
         "xla_sharding_util.cpp",
+        "xla_manual_registration.cpp",
         ":RegisterAutogradXLA.cpp",
         ":RegisterXLA.cpp",
         ":XLANativeFunctions.cpp",

diff --git a/torch_xla/csrc/nms_op.cpp b/torch_xla/csrc/nms_op.cpp
diff --git a/torch_xla/csrc/nms_op.h b/torch_xla/csrc/nms_op.h