PR #6814 Refactor `nms` into TorchVision variant.

Refactor `nms` into TorchVision variant. #6814

ysiraichi merged 7 commits into master from ysiraichi/refactor-into-torchvision-nms

ysiraichi requested a review from

JackCaoG 1 year ago

ysiraichi added xla:gpu

Kill the old `nms` function.

ce758e76

Refactor and implement new `nms`.

4fd57d75

Add tests.

cb29d771

Fix lint issue.

edc269a4

Skip functionalization when returning.

be218314

ysiraichi force pushed from a4e90c80 to be218314 1 year ago

Fix test.

33e5760e

Fix lint.

cfeb34d4

vanbasten23 commented on 2024-03-27

vanbasten23 commented on 2024-03-29

vanbasten23 approved these changes on 2024-03-29

ysiraichi merged 6cf9b915 into master 1 year ago

miladm commented on 2024-04-08

Reviewers

vanbasten23

miladm

JackCaoG

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone