| .. |
|
batch_permutation_op.cc
|
Tensor construction codemod(ResizeLike) - 7/7 (#15087)
|
2018-12-20 15:33:07 -08:00 |
|
batch_permutation_op.cu
|
|
|
|
batch_permutation_op.h
|
|
|
|
CMakeLists.txt
|
Reenable OpenMP by reverting the following two commits. (#15315)
|
2018-12-17 19:54:41 -08: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
|
Tensor method rename dims()->sizes() - 1/2
|
2019-01-04 16:02:22 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08:00 |
|
sample_as_op.h
|
|
|
|
select_smooth_l1_loss_op.cc
|
|
|
|
select_smooth_l1_loss_op.cu
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Tensor construction codemod - 2/2 (#15600)
|
2019-01-04 13:31:53 -08: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
|
Remove many caffe2::TIndex and replace them with int64_t (#11943)
|
2018-09-22 18:11:04 -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
|
Fix UpsampleNearest op CPU impl batch handling (#13002)
|
2018-10-24 13:10:53 -07:00 |