projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: a47749c) | patch
Adding CUDA version for C2 operators generate proposals and nms (#13694)
authorhbraun@nvidia.com <hbraun@nvidia.com>
Thu, 20 Dec 2018 22:24:27 +0000 (14:24 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 20 Dec 2018 22:39:09 +0000 (14:39 -0800)
commit3fdf567752b484c5542efea02afc64e4f875eeca
tree916d50564305376b5540584338011264b989c0cdtree | snapshot
parenta47749cb28bb630bc99601a64e24ad453d9583a9commit | diff
Adding CUDA version for C2 operators generate proposals and nms (#13694)

Summary:
Related to issue #13684
Pull Request resolved: https://github.com/pytorch/pytorch/pull/13694

Reviewed By: wat3rBro

Differential Revision: D13017791

Pulled By: newstzpz

fbshipit-source-id: 4bdc58e474d8e1f6cd73a02bf51f91542a2b9d0b
caffe2/core/common_gpu.h diff | blob | history
caffe2/operators/generate_proposals_op.cc diff | blob | history
caffe2/operators/generate_proposals_op.cu [new file with mode: 0644] blob
caffe2/operators/generate_proposals_op.h diff | blob | history
caffe2/operators/generate_proposals_op_gpu_test.cc [new file with mode: 0644] blob
caffe2/operators/generate_proposals_op_util_nms_gpu.cu [new file with mode: 0644] blob
caffe2/operators/generate_proposals_op_util_nms_gpu.h [new file with mode: 0644] blob
caffe2/operators/generate_proposals_op_util_nms_gpu_test.cc [new file with mode: 0644] blob
tools/amd_build/pyHIPIFY/cuda_to_hip_mappings.py diff | blob | history
Domain: Machine Learning / ML Framework;