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
Viswanath Sivakumar 1bea5fc3ad Fix UpsampleNearest op CPU impl batch handling (#13002) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/13002

Batch dim wasn't handled in the CPU impl (will fail for inputs with N > 1).
Fixing that here.

Differential Revision: D10515159

fbshipit-source-id: ee7e4f489d2d4de793f550b31db7c0e2ba3651e8
2018-10-24 13:10:53 -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
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
group_spatial_softmax_op.cu
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
ps_roi_pool_op.cu Move the location of annotation 2018-10-24 12:35:08 -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
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
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
select_smooth_l1_loss_op.cc
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
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
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
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
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
spatial_narrow_as_op.cu
spatial_narrow_as_op.h
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 Fix UpsampleNearest op CPU impl batch handling (#13002) 2018-10-24 13:10:53 -07:00
upsample_nearest_op_test.py Fix UpsampleNearest op CPU impl batch handling (#13002) 2018-10-24 13:10:53 -07:00