Expose cuda kernel for caffe2::GenerateProposals
authorSebastian Messmer <messmer@fb.com>
Mon, 4 Mar 2019 22:53:55 +0000 (14:53 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 4 Mar 2019 22:59:08 +0000 (14:59 -0800)
commit910519e45b8917e6e3a9aadc0f9e42f91cc6a1bf
treea0143bc0ab5591af9295bc05b04aabb4c662cd21
parentaea8dd8377da45a2c51d02ddaa7846d72cde6966
Expose cuda kernel for caffe2::GenerateProposals

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

Reviewed By: ezyang, wat3rBro

Differential Revision: D14071130

fbshipit-source-id: 6fe26503f6069c36ec31d6c09b549b932d5db242
caffe2/operators/generate_proposals_op.cu
caffe2/operators/generate_proposals_op.h
caffe2/python/operator_test/torch_integration_test.py