saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/modules/detectron
History
wuhuikx e497aa1e35 Optimize UpsampleNearest Op (#12151) 
Summary:
Optimize the UpsampleNearest Op.
1. Add OMP
2. revise the translated_idx method
Pull Request resolved: https://github.com/pytorch/pytorch/pull/12151

Differential Revision: D10362856

Pulled By: ezyang

fbshipit-source-id: 535a4b87c7423942217f2d79bedc463a0617c67a
2018-10-16 20:34:20 -07:00
..
batch_permutation_op.cc Add the OpenMP optimization for BatchPermutation. (#12153) 2018-10-16 20:23:09 -07:00
batch_permutation_op.cu Import Detectron ops 2018-01-17 10:31:30 -08:00
batch_permutation_op.h [Caffe2] Detectron fpn support (#6645) 2018-04-18 10:00:49 -07:00
CMakeLists.txt Don't build Detectron ops with NO_CAFFE2_OPS=1 (#11799) 2018-09-18 14:09:33 -07:00
group_spatial_softmax_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
group_spatial_softmax_op.cu Import Detectron ops 2018-01-17 10:31:30 -08:00
group_spatial_softmax_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
ps_roi_pool_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
ps_roi_pool_op.cu Update from facebook (#6871) 2018-04-23 15:01:56 -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 Update from facebook (#6871) 2018-04-23 15:01:56 -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 Make caffe2::Tensor::dims() return an IntList instead of a const vector& (#12180) 2018-10-05 15:57:41 -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 Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -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 Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
sigmoid_cross_entropy_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
sigmoid_focal_loss_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
sigmoid_focal_loss_op.cu Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -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 Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -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 Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
softmax_focal_loss_op.h Workaround for Cuda9.2 and GCC7 compilation errors (#10510) 2018-08-14 20:54:52 -07:00
spatial_narrow_as_op.cc Import Detectron ops 2018-01-17 10:31:30 -08:00
spatial_narrow_as_op.cu Import Detectron ops 2018-01-17 10:31:30 -08:00
spatial_narrow_as_op.h Import Detectron ops 2018-01-17 10:31:30 -08:00
upsample_nearest_op.cc Enable Detectron model inference for CPU and MKL-DNN paths (#10157) 2018-08-29 15:11:01 -07:00
upsample_nearest_op.cu Remove many caffe2::TIndex and replace them with int64_t (#11943) 2018-09-22 18:11:04 -07:00
upsample_nearest_op.h Optimize UpsampleNearest Op (#12151) 2018-10-16 20:34:20 -07:00