frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
6af248261295977d882f2cc66cc294026b66725c
pytorch/modules/detectron
History
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
[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
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
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
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
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