pytorch/modules/detectron
Xiaomeng Yang 821b5f138a Optimize SoftmaxOp on CPU (#18635)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/18635

Optimize SoftmaxOp on CPU

Reviewed By: houseroad

Differential Revision: D14689516

fbshipit-source-id: d2dcee2476d1a3a21f428e99bce9835f1d229d64
2019-04-10 18:52:15 -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
CMakeLists.txt Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
group_spatial_softmax_op.cc Optimize SoftmaxOp on CPU (#18635) 2019-04-10 18:52:15 -07: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
ps_roi_pool_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
ps_roi_pool_op.h
roi_pool_f_op.cc
roi_pool_f_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
roi_pool_f_op.h
sample_as_op.cc
sample_as_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
sample_as_op.h
select_smooth_l1_loss_op.cc
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
sigmoid_cross_entropy_loss_op.cc
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
sigmoid_focal_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
sigmoid_focal_loss_op.h
smooth_l1_loss_op.cc
smooth_l1_loss_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
smooth_l1_loss_op.h
softmax_focal_loss_op.cc Optimize SoftmaxOp on CPU (#18635) 2019-04-10 18:52:15 -07: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
spatial_narrow_as_op.cu Enable detectron on AMD GPU 2019-03-12 16:29:42 -07:00
spatial_narrow_as_op.h
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
upsample_nearest_op_test.py