register BoxWithNMSLimit with C10
authorYanghan Wang <yanghan@instagram.com>
Fri, 29 Mar 2019 20:31:45 +0000 (13:31 -0700)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 29 Mar 2019 20:41:40 +0000 (13:41 -0700)
commitf4e35d30ed32d9d948d002eb6bc9e235fb433fb7
tree72f15fa4e960410e6afe07e2e66ea1028c5e0501
parentd895d30876015862ba64e40fde610274c281f507
register BoxWithNMSLimit with C10

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17956

Reviewed By: houseroad

Differential Revision: D14417300

fbshipit-source-id: eb5e2ba84513b3b7bfa509dc442424b13fe9148f
caffe2/operators/box_with_nms_limit_op.cc
caffe2/operators/box_with_nms_limit_op.h
caffe2/python/operator_test/torch_integration_test.py