OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
173f224570
pytorch/modules/detectron
History
Sandeep Kumar 6248266d91 Enable detectron on AMD GPU 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/17862

Differential Revision: D14429234

Pulled By: bddppq

fbshipit-source-id: 5cb8750bd9db0ff8a179977d2bfbb180265cce81
2019-03-12 16:29:42 -07:00
..
batch_permutation_op.cc Tensor construction codemod(ResizeLike) - 7/7 (#15087) 2018-12-20 15:33:07 -08:00
batch_permutation_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
batch_permutation_op.h [Caffe2] Detectron fpn support (#6645) 2018-04-18 10:00:49 -07:00
CMakeLists.txt Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
group_spatial_softmax_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
group_spatial_softmax_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
group_spatial_softmax_op.h Tensor reinitialization codemod - 5/5 (#15884) 2019-01-10 16:32:26 -08:00
ps_roi_pool_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
ps_roi_pool_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
ps_roi_pool_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
roi_pool_f_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
roi_pool_f_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
roi_pool_f_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
sample_as_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
sample_as_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
sample_as_op.h Import Detectron ops 2018-01-17 10:31:30 -08:00
select_smooth_l1_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
select_smooth_l1_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
select_smooth_l1_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
sigmoid_cross_entropy_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
sigmoid_cross_entropy_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
sigmoid_cross_entropy_loss_op.h Tensor reinitialization codemod - 5/5 (#15884) 2019-01-10 16:32:26 -08:00
sigmoid_focal_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
sigmoid_focal_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
sigmoid_focal_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
smooth_l1_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
smooth_l1_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
smooth_l1_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
softmax_focal_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
softmax_focal_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
softmax_focal_loss_op.h Tensor reinitialization codemod - 5/5 (#15884) 2019-01-10 16:32:26 -08:00
spatial_narrow_as_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
spatial_narrow_as_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
spatial_narrow_as_op.h Import Detectron ops 2018-01-17 10:31:30 -08:00
upsample_nearest_op_test.py Fix UpsampleNearest op CPU impl batch handling (#13002) 2018-10-24 13:10:53 -07:00
upsample_nearest_op.cc seperate mkl, mklml, and mkldnn (#12170) 2018-10-29 10:52:55 -07:00
upsample_nearest_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
upsample_nearest_op.h Tensor construction codemod - 3/3 (#14835) 2018-12-06 11:50:59 -08:00