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
26740f853e10e34a7e8d2b72b950875796350418
pytorch/caffe2/operators
History
Eddie Yan 967dd31621 [cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862) 
Follow-up of #95722

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120862
Approved by: https://github.com/Skylion007
2024-03-07 01:46:25 +00:00
..
hip
[ROCm] Update HIP_VERSION to TORCH_HIP_VERSION (#62786)
2021-08-13 15:00:43 -07:00
quantized
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
rnn
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
abs_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
abs_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
abs_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
accumulate_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
accumulate_op.cu
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
accumulate_op.h
refactor caffe2 operator constructors - 1/9 (#17082)
2019-03-04 16:04:01 -08:00
accuracy_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
accuracy_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
accuracy_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
acos_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
acos_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
acos_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
activation_ops_cudnn.h
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
affine_channel_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
affine_channel_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
affine_channel_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
alias_with_name.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
alias_with_name.cu
move AliasWithNameOp to caffe2/operators
2019-12-17 02:39:40 -08:00
alias_with_name.h
move AliasWithNameOp to caffe2/operators
2019-12-17 02:39:40 -08:00
apmeter_op.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
apmeter_op.h
refactor caffe2 operator constructors - 1/9 (#17082)
2019-03-04 16:04:01 -08:00
arg_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
arg_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
arg_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
asin_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
asin_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
asin_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
assert_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
assert_op.cu
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
assert_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
async_net_barrier_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
async_net_barrier_op.cu
[DPER] Introduce barrier operation to force synchronization of threads in async execution (#49322)
2020-12-15 16:13:42 -08:00
async_net_barrier_op.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
atan_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
atan_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
atan_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
atomic_ops.cc
fix torch.config can't respect USE_MKLDNN flag issue (#75001)
2022-07-17 15:00:48 +00:00
batch_box_cox_op.cc
[torch] Unify batch_box_cox implementations into perfkernels folder (#86569)
2022-10-23 19:29:25 +00:00
batch_box_cox_op.h
[torch] Unify batch_box_cox implementations into perfkernels folder (#86569)
2022-10-23 19:29:25 +00:00
batch_bucketize_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
batch_bucketize_op.h
refactor caffe2 operator constructors - 1/9 (#17082)
2019-03-04 16:04:01 -08:00
batch_gather_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
batch_gather_ops.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
batch_gather_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
batch_matmul_op_gpu_test.cc
Renaming size() to numel() - 1/6
2018-10-29 11:11:19 -07:00
batch_matmul_op_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
batch_matmul_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
batch_matmul_op.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
batch_matmul_op.h
Optimize batch mm op when broadcast the second input (#21556)
2019-06-09 15:28:03 -07:00
batch_moments_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
batch_moments_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
batch_moments_op.h
refactor caffe2 operator constructors - 10/9 (#17659)
2019-03-06 15:11:47 -08:00
batch_permutation_op_gpu_test.cc
Add zero input support for batch permutation op (#39851)
2020-06-13 21:34:24 -07:00
batch_permutation_op.cc
fix torch.config can't respect USE_MKLDNN flag issue (#75001)
2022-07-17 15:00:48 +00:00
batch_permutation_op.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
batch_permutation_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
batch_sparse_to_dense_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
batch_sparse_to_dense_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
batch_sparse_to_dense_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
bbox_transform_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
bbox_transform_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
bisect_percentile_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
bisect_percentile_op.h
Clean up bisect_percentile_op (#73148)
2022-02-25 04:33:45 +00:00
boolean_mask_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
boolean_mask_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
boolean_mask_ops.h
Adding gradient to Boolean Mask operator (#21423)
2019-06-06 20:48:47 -07:00
boolean_unmask_ops_test.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
boolean_unmask_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
boolean_unmask_ops.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
boolean_unmask_ops.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
box_with_nms_limit_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
box_with_nms_limit_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
bucketize_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
bucketize_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
bucketize_op.h
[pyper] export caffe2 bucketize GPU operator to pytorch
2020-09-09 16:08:53 -07:00
byte_weight_dequant_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
byte_weight_dequant_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
cast_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
cast_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
cast_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
cbrt_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
cbrt_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
cbrt_op.h
Use FixedDivisor in Reduce and Broadcast CUDA kernels (#9072)
2018-07-01 00:25:34 -07:00
cc_bmm_bg_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
cc_bmm_bg_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
ceil_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
ceil_op.cu
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
ceil_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
channel_backprop_stats_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
channel_backprop_stats_op.cu
Drop more unused variables (#71123)
2022-01-11 12:46:24 -08:00
channel_backprop_stats_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
channel_shuffle_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
channel_shuffle_op.cu
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
channel_shuffle_op.h
refactor caffe2 operator constructors - 11/9 (#17722)
2019-03-08 12:38:54 -08:00
channel_stats_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
channel_stats_op.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
channel_stats_op.h
Optimize channel_stats_op (#16243)
2019-03-12 12:08:00 -07:00
channelwise_conv3d_op_cudnn.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
clip_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
clip_op.cu
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
clip_op.h
refactor caffe2 operator constructors - 2/9 (#17083)
2019-02-28 14:23:55 -08:00
CMakeLists.txt
Remove experimental c10 ops (#36394)
2020-04-10 16:11:16 -07:00
collect_and_distribute_fpn_rpn_proposals_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
collect_and_distribute_fpn_rpn_proposals_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
communicator_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
communicator_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
concat_split_op_gpu.cc
Update from facebook (#7451)
2018-05-10 23:14:27 -07:00
concat_split_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
concat_split_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
conditional_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
conditional_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
conv_gradient_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
conv_op_cache_cudnn_test.cc
move flags to c10 (#12144)
2018-10-04 02:09:56 -07:00
conv_op_cache_cudnn.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
conv_op_cache_cudnn.h
Rename IntList to IntArrayRef. (#16751)
2019-02-05 14:54:34 -08:00
conv_op_cudnn.cc
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
conv_op_eigen.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
conv_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
conv_op_impl.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
conv_op_shared_gpu.cc
Removing some dependency edges from Blob to other caffe2 (#12043)
2018-09-25 11:40:24 -07:00
conv_op_shared.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
conv_op_shared.h
Remove template parameter from Tensor (#9939)
2018-07-27 10:56:39 -07:00
conv_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
conv_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
conv_pool_op_base.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
conv_transpose_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
conv_transpose_op_cudnn.cc
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
conv_transpose_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
conv_transpose_op_impl.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
conv_transpose_op_mobile_impl.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
conv_transpose_op_mobile_test.cc
[clang-tidy] Exclude cppcoreguidelines-avoid-magic-numbers (#57841)
2021-05-07 20:02:33 -07:00
conv_transpose_op_mobile.cc
Make C10_MOBILE consistent with how feature macros are usually used (#17481)
2019-02-27 17:57:51 -08:00
conv_transpose_op_mobile.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
conv_transpose_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
conv_transpose_op.h
refactor caffe2 operator constructors - 2/9 (#17083)
2019-02-28 14:23:55 -08:00
conv_transpose_unpool_op_base.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
copy_op.cc
[caffe2] Fix alias analysis for quantization compression ops (#74169)
2022-03-15 22:29:59 +00:00
copy_op.cu
exposing CPU/GPU Copy ops (#32248)
2020-01-17 12:40:43 -08:00
copy_op.h
exposing CPU/GPU Copy ops (#32248)
2020-01-17 12:40:43 -08:00
copy_rows_to_tensor_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
copy_rows_to_tensor_op.h
Make PyTorch code-base clang-tidy compliant (#56892)
2021-04-28 14:10:25 -07:00
cos_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
cos_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
cos_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
cosh_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
cosh_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
cosh_op.h
Implement Sinh and Cosh (#9213)
2018-07-10 18:55:31 -07:00
cosine_embedding_criterion_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
cosine_embedding_criterion_op.cu
Forbid trailing whitespace (#53406)
2021-03-05 17:22:55 -08:00
cosine_embedding_criterion_op.h
refactor caffe2 operator constructors - 2/9 (#17083)
2019-02-28 14:23:55 -08:00
counter_ops_gpu.cc
HIP Operators Generator--> HipOpG (#9322)
2018-07-19 00:26:06 -07:00
counter_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
counter_ops.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
crash_op.cc
Support compilation on gcc-7.4.0 (#19470)
2019-04-19 21:41:36 -07:00
create_scope_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
create_scope_op.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
crf_viterbi_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
cross_entropy_op.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
cross_entropy_op.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
cross_entropy_op.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
ctc_beam_search_decoder_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
ctc_beam_search_decoder_op.h
Output sequence probability with CTC beam search, optional multiple output sequences (#21927)
2019-07-02 17:29:13 -07:00
ctc_greedy_decoder_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
ctc_greedy_decoder_op.h
refactor caffe2 operator constructors - 2/9 (#17083)
2019-02-28 14:23:55 -08:00
cube_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
cube_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
cube_op.h
Add Cube and Cbrt Ops (#8991)
2018-06-28 14:55:30 -07:00
data_couple_gpu.cu
No Op Optimizer (#12390)
2018-10-10 18:09:46 -07:00
data_couple.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
data_couple.h
No Op Optimizer (#12390)
2018-10-10 18:09:46 -07:00
dataset_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
dataset_ops.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
deform_conv_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
deform_conv_op_impl.h
Tentative fix for CUDA-10.2 windows build failures (#76204)
2022-04-22 15:15:18 +00:00
deform_conv_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
deform_conv_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
deform_conv_op.h
refactor caffe2 operator constructors - 2/9 (#17083)
2019-02-28 14:23:55 -08:00
dense_vector_to_id_list_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
dense_vector_to_id_list_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
depthwise_3x3_conv_op_cudnn.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
distance_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
distance_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
distance_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
do_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
do_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
do_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
dropout_op_cudnn.cc
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
dropout_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
dropout_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
dropout_op.h
[caffe2] allow dropout to take 1.0 as dropout ratio to zero-out a layer (#72741)
2022-02-15 19:14:46 +00:00
elementwise_add_gradient_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_add_op_gpu.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_add_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_add_op.h
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_div_gradient_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_div_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_div_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
elementwise_div_op.h
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_linear_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
elementwise_linear_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
elementwise_linear_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
elementwise_logical_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
elementwise_logical_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
elementwise_mul_gradient_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_mul_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_mul_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
elementwise_mul_op.h
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_op_gpu_test.cc
move flags to c10 (#12144)
2018-10-04 02:09:56 -07:00
elementwise_op_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
elementwise_op_test.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
elementwise_ops_schema.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
elementwise_ops_utils.cc
Make PyTorch code-base clang-tidy compliant (#56892)
2021-04-28 14:10:25 -07:00
elementwise_ops_utils.h
[PyPer] Port c2 add to pt (#54229)
2021-03-19 12:45:11 -07:00
elementwise_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
elementwise_ops.cu
[ROCm] more 6.0 changes (#115946)
2023-12-20 20:19:29 +00:00
elementwise_ops.h
[efficiency_camp] Vector Realloc Optimize caffe2::BinaryElementwiseWithArgsOp::DoRunWithType (#100631)
2023-05-10 23:34:54 +00:00
elementwise_sub_gradient_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_sub_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
elementwise_sub_op.cc
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_sub_op.h
make broadcast fastpath the default for currently rolled-out ops (#68365)
2021-11-15 21:41:57 -08:00
elementwise_sum_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
elu_op_cudnn.cc
refactor caffe2 operator constructors - 3/9 (#17084)
2019-02-28 14:13:17 -08:00
elu_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
elu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
elu_op.h
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
enforce_finite_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
enforce_finite_op.cu
Remove calls to CopyFrom that can be sync (#13205)
2018-10-29 13:57:42 -07:00
enforce_finite_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
ensure_clipped_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
ensure_clipped_op.h
refactor caffe2 operator constructors - 3/9 (#17084)
2019-02-28 14:13:17 -08:00
ensure_cpu_output_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
ensure_cpu_output_op.cu
Update from facebook (#7451)
2018-05-10 23:14:27 -07:00
ensure_cpu_output_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
erf_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
erf_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
erf_op.h
Export PyTorch erf to ONNX Erf and add Caffe2 Erf operator
2019-01-17 09:18:08 -08:00
exp_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
exp_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
exp_op.h
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
expand_op_gpu.cc
New operator 'expand' (#8263)
2018-06-18 16:33:47 -07:00
expand_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
expand_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
expand_squeeze_dims_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
expand_squeeze_dims_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
expand_squeeze_dims_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
fc_inference.cc
Fix bytes_written and bytes_read (#64244)
2021-09-01 13:35:41 -07:00
fc_inference.h
[caffe2] add cost inference for FusedFakeQuantFC and FusedFakeQuantFCGradient (#44840)
2020-09-17 14:07:17 -07:00
feature_maps_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
feature_maps_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
feed_blob_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
feed_blob_op.h
refactor caffe2 operator constructors - 3/9 (#17084)
2019-02-28 14:13:17 -08:00
filler_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
filler_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
filler_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
find_duplicate_elements_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
find_duplicate_elements_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
find_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
find_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
find_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
flatten_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
flatten_op.h
Revert "Tensor construction codemod(raw_mutable_data) (#16373)" (#18680)
2019-04-01 14:39:13 -07:00
flexible_top_k.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
flexible_top_k.h
refactor caffe2 operator constructors - 3/9 (#17084)
2019-02-28 14:13:17 -08:00
floor_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
floor_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
floor_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
free_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
free_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
free_op.h
refactor caffe2 operator constructors - 3/9 (#17084)
2019-02-28 14:13:17 -08:00
fully_connected_op_gpu.cc
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
fully_connected_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
fully_connected_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
fused_rowwise_8bit_conversion_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
fused_rowwise_8bit_conversion_ops.h
Remove little endian asserts (#99713)
2023-04-26 02:08:28 +00:00
fused_rowwise_nbit_conversion_ops.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
fused_rowwise_nbit_conversion_ops.h
Remove little endian asserts (#99713)
2023-04-26 02:08:28 +00:00
fused_rowwise_nbitfake_conversion_ops.cc
Compile without -Wno-unused-variable (take 2) (#66041)
2021-10-04 20:39:39 -07:00
fused_rowwise_nbitfake_conversion_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
fused_rowwise_random_quantization_ops.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
fused_rowwise_random_quantization_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
gather_fused_8bit_rowwise_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
gather_fused_8bit_rowwise_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
gather_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
gather_op.cu
support Gather different indices for different examples in one batch (#23813)
2019-08-07 21:14:30 -07:00
gather_op.cuh
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
gather_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
gather_ranges_to_dense_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
gather_ranges_to_dense_op.h
[Resubmit][S372460 follow up] Reduce embedding feature validation failure carry-on impact (#111838)
2023-10-28 03:50:33 +00:00
gelu_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
gelu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
gelu_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
generate_proposals_op_gpu_test.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
generate_proposals_op_test.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
generate_proposals_op_util_boxes_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
generate_proposals_op_util_boxes.h
[codemod] Fix shadows in PyTorch (#117562)
2024-01-17 20:33:50 +00:00
generate_proposals_op_util_nms_gpu_test.cc
Check all CUDA API calls for errors in caffe2/ (#81816)
2022-10-28 00:41:06 +00:00
generate_proposals_op_util_nms_gpu.cu
Check all CUDA API calls for errors in caffe2/ (#81816)
2022-10-28 00:41:06 +00:00
generate_proposals_op_util_nms_gpu.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
generate_proposals_op_util_nms_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
generate_proposals_op_util_nms.h
[EZ][BE] Fix the massively annoying strict-weak-ordering issue. (#105189)
2023-07-19 19:37:50 +00:00
generate_proposals_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
generate_proposals_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
generate_proposals_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
given_tensor_byte_string_to_uint8_fill_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
given_tensor_byte_string_to_uint8_fill_op.cu
Export uint8 tensors as byte string in mobile_exporter and add GivenTensorByteStringToUInt8FillOp (#10385)
2018-08-15 14:26:50 -07:00
given_tensor_byte_string_to_uint8_fill_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
given_tensor_fill_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
given_tensor_fill_op.cu
Add GivenTensorInt16Fill (#20515)
2019-05-15 19:45:15 -07:00
given_tensor_fill_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
glu_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
glu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
glu_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
group_norm_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
group_norm_op.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
group_norm_op.h
batch size 0 support in norm operators (#26894)
2019-09-26 16:08:35 -07:00
gru_unit_op_gpu.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
gru_unit_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
gru_unit_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
h_softmax_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
h_softmax_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
half_float_ops_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
half_float_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
half_float_ops.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
half_float_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
hard_sigmoid_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
hard_sigmoid_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
hard_sigmoid_op.h
Add CPU version of hard sigmoid operator to caffe2 (#10837)
2018-08-28 14:55:49 -07:00
heatmap_max_keypoint_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
heatmap_max_keypoint_op.h
register HeatmapMaxKeypoint with C10 (#25191)
2019-08-27 20:13:57 -07:00
histogram_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
histogram_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
if_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
if_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
if_op.h
refactor caffe2 operator constructors - 4/9 (#17085)
2019-02-28 14:23:52 -08:00
im2col_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
im2col_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
im2col_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
index_hash_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
index_hash_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
index_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
index_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
inference_lstm_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
inference_lstm_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
instance_norm_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
instance_norm_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
instance_norm_op.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
instance_norm_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
integral_image_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
integral_image_op.cu
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
integral_image_op.h
refactor caffe2 operator constructors - 4/9 (#17085)
2019-02-28 14:23:52 -08:00
is_empty_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
is_empty_op.h
Tensor construction: combine Resize+mutable_data - 2/4 (#14205)
2018-11-30 10:46:58 -08:00
jsd_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
jsd_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
key_split_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
key_split_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
last_n_window_collector.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
layer_norm_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
layer_norm_op.cu
Drop unused variables and add some const (#71106)
2022-01-11 12:38:59 -08:00
layer_norm_op.h
Compile without -Wno-unused-variable (take 2) (#66041)
2021-10-04 20:39:39 -07:00
leaky_relu_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
leaky_relu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
leaky_relu_op.h
[codemod][caffe2] Run clang-format - 5/7
2020-06-30 15:45:11 -07:00
length_split_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
length_split_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
lengths_pad_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_pad_op.cu
Update from facebook (#6871)
2018-04-23 15:01:56 -07:00
lengths_pad_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
lengths_reducer_fused_8bit_rowwise_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_reducer_fused_8bit_rowwise_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
lengths_reducer_fused_nbit_rowwise_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_reducer_fused_nbit_rowwise_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
lengths_reducer_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_reducer_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
lengths_reducer_rowwise_8bit_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_reducer_rowwise_8bit_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
lengths_tile_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_tile_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
lengths_tile_op.h
Tensor reinitialization codemod - 3/5 (#15912)
2019-01-16 19:49:01 -08:00
lengths_top_k_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lengths_top_k_op.h
refactor caffe2 operator constructors - 4/9 (#17085)
2019-02-28 14:23:52 -08:00
listwise_l2r_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
listwise_l2r_op.h
add LambdaRank DCG Loss Option (#23679)
2019-08-02 11:47:46 -07:00
load_save_op_gpu.cc
Rename cuda_gpu_id to device_id in DeviceOption (#12456)
2018-10-09 15:54:04 -07:00
load_save_op_util.cc
Reduce amount of work done within a global lock within ParallelLoadOp (#43508)
2020-08-26 18:19:40 -07:00
load_save_op_util.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
load_save_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
load_save_op.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
local_response_normalization_op_cudnn.cc
Compile without -Wno-unused-variable (take 2) (#66041)
2021-10-04 20:39:39 -07:00
local_response_normalization_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
local_response_normalization_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
local_response_normalization_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
locally_connected_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
locally_connected_op_impl.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
locally_connected_op_util.cc
Update Im2ColNd functions (#7505)
2018-05-12 15:59:50 -07:00
locally_connected_op_util.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
locally_connected_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
locally_connected_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
log1p_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
log1p_op.cu
[caffe2] Reintroduce Log1p operator (#55073)
2021-03-31 22:29:23 -07:00
log1p_op.h
[caffe2] Reintroduce Log1p operator (#55073)
2021-03-31 22:29:23 -07:00
log_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
log_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
log_op.h
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
logit_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
logit_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
logit_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
loss_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
loss_op.cu
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
loss_op.h
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
lp_pool_op.cc
Fix typo under caffe2 directory (#110825)
2023-10-08 20:48:12 +00:00
lp_pool_op.cu
Fix unused variable warnings (#72410)
2022-02-07 18:25:29 +00:00
lpnorm_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
lpnorm_op.cu
Back out "Back out "[c2] register cuda op for LpNorm (fallback)"" (#38566)
2020-05-19 10:37:25 -07:00
lpnorm_op.h
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
lstm_unit_op_gpu.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
lstm_unit_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
lstm_unit_op.h
[caffe2] Explicit vectorization of LSTM operator (#35556)
2020-04-01 17:19:56 -07:00
lstm_utils.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
map_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
map_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
margin_loss_l2r_op.cc
[1/n][caffe2] Add session based margin loss function in caffe2 operator
2022-01-21 23:13:36 +00:00
margin_loss_l2r_op.h
[1/n][caffe2] Add session based margin loss function in caffe2 operator
2022-01-21 23:13:36 +00:00
margin_ranking_criterion_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
margin_ranking_criterion_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
margin_ranking_criterion_op.h
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
matmul_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
matmul_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
matmul_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
max_pool_with_index_gpu.h
HIP Operators Generator--> HipOpG (#9322)
2018-07-19 00:26:06 -07:00
max_pool_with_index.cu
Avoid type qualifier specified more than once (#72411)
2022-02-07 18:25:29 +00:00
mean_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
mean_op.cu
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
mean_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
mem_query_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
merge_id_lists_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
merge_id_lists_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
minmax_gradient_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
minmax_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
minmax_ops.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
minmax_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
mish_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
mish_op.h
[Caffe2] Optimize MishOp on CPU (#48212)
2020-11-19 14:17:27 -08:00
mod_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
mod_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
mod_op.h
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
moments_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
moments_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
moments_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
multi_class_accuracy_op.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
multi_class_accuracy_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
multi_class_accuracy_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
negate_gradient_op_gpu.cc
NegateGradientOp and test
2017-11-13 18:05:14 -08:00
negate_gradient_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
negate_gradient_op.h
Shut up "address will always evaluate to 'true'" warnings (#14774)
2018-12-05 21:18:31 -08:00
negative_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
negative_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
negative_op.h
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
ngram_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
ngram_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
no_default_engine_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
norm_planar_yuv_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
normalize_l1_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
normalize_l1_op.h
Revert D22330340: [C2] Fixed a bug in normalization operator
2020-07-02 16:05:23 -07:00
normalize_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
normalize_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
normalize_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
numpy_tile_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
numpy_tile_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
one_hot_ops.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
one_hot_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
one_hot_ops.h
Export BatchBucketOneHot Caffe2 Operator to PyTorch
2020-08-11 14:00:19 -07:00
onnx_while_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
onnx_while_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
op_utils_cudnn.h
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
operator_fallback_gpu_test.cc
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
operator_fallback_gpu.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
order_switch_ops_cudnn.cc
Reapply D14078519 (#17596)
2019-03-06 13:51:00 -08:00
order_switch_ops_gpu.cc
Optimize NCHW2NHWC on GPU (#12910)
2018-10-22 11:24:29 -07:00
order_switch_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
order_switch_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
pack_rnn_sequence_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
pack_rnn_sequence_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
pack_segments.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
pack_segments.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
pack_segments.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
pad_op_gpu.cu
Drop unused variables and add some const (#71106)
2022-01-11 12:38:59 -08:00
pad_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
pad_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
partition_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
partition_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
percentile_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
percentile_op.h
Exposing Percentile Caffe2 Operator in PyTorch
2020-08-07 16:22:37 -07:00
perplexity_op.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
perplexity_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
perplexity_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
piecewise_linear_transform_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
piecewise_linear_transform_op.cu
Do not use thrust::lower_bound on device (#80746)
2022-07-02 03:00:27 +00:00
piecewise_linear_transform_op.h
Tentative fix for CUDA-10.2 windows build failures (#76204)
2022-04-22 15:15:18 +00:00
pool_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
pool_op_cudnn.cc
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
pool_op_util.cc
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
pool_op_util.h
Add count_include_pad arg for PoolOpGradient on CPU and fix ARM performance issue. (#15651)
2019-01-03 00:18:47 -08:00
pool_op.cc
Fix typo under caffe2 directory (#110825)
2023-10-08 20:48:12 +00:00
pool_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
pool_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
pow_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
pow_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
pow_op.h
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
prefetch_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
prelu_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
prelu_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
prelu_op.h
refactor caffe2 operator constructors - 6/9 (#17087)
2019-02-28 14:23:57 -08:00
prepend_dim_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
prepend_dim_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
prepend_dim_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
quant_decode_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
quant_decode_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
quantile_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
quantile_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
rank_loss_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
rank_loss_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
reciprocal_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
reciprocal_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reciprocal_op.cu
[ONNX] Use Reciprocal operator instead of Div(1, x). (#65382) (#67271)
2021-10-28 08:01:21 -07:00
reciprocal_op.h
Adding reciprocal operator and a test
2018-07-27 18:24:43 -07:00
reduce_front_back_max_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reduce_front_back_max_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
reduce_front_back_max_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
reduce_front_back_mean_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reduce_front_back_sum_mean_ops.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
reduce_front_back_sum_mean_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
reduce_front_back_sum_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reduce_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reduce_ops.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
reduce_ops.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
reducer_functors.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
reduction_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reduction_ops.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
reduction_ops.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
relu_n_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
relu_n_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
relu_n_op.h
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
relu_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
relu_op.cu
[ROCm] Update HIP_VERSION to TORCH_HIP_VERSION (#62786)
2021-08-13 15:00:43 -07:00
relu_op.h
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
remove_data_blocks_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
remove_data_blocks_op.h
use irange for loops 6 (#66742)
2021-12-07 16:07:50 -08:00
replace_nan_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
replace_nan_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
replace_nan_op.h
refactor caffe2 operator constructors - 7/9 (#17088)
2019-02-28 14:23:53 -08:00
reservoir_sampling.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
reshape_op_gpu_test.cc
Rename ndim() -> dim() - 5/6
2018-11-06 16:38:35 -08:00
reshape_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
reshape_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
reshape_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
resize_3d_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
resize_3d_op.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
resize_3d_op.h
Migrate the cpu and gpu implementations of resize nearest 3D from vision to caffe2
2019-10-03 16:14:00 -07:00
resize_op.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
resize_op.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
resize_op.h
Rename caffe2<->c10 operator wrappers (#21322)
2019-06-07 13:48:10 -07:00
reverse_packed_segs_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
reverse_packed_segs_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
reverse_packed_segs_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
rmac_regions_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
rmac_regions_op.cu
Refactor cub namespace handling (#66219)
2021-10-25 14:37:09 -07:00
rmac_regions_op.h
refactor caffe2 operator constructors - 7/9 (#17088)
2019-02-28 14:23:53 -08:00
rms_norm_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
rms_norm_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
rms_norm_op.h
[Caffe2] Add RMSNormOp (#44338)
2020-09-08 23:50:44 -07:00
roi_align_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
roi_align_gradient_op.cu
Drop unused variables and add some const (#71106)
2022-01-11 12:38:59 -08:00
roi_align_gradient_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
roi_align_op_gpu_test.cc
Re-enable Caffe2 test RoiAlignTest.CheckCPUGPUEqual (#40901)
2020-07-02 11:22:23 -07:00
roi_align_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
roi_align_op.cu
Drop more unused variables (#71123)
2022-01-11 12:46:24 -08:00
roi_align_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
roi_align_rotated_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
roi_align_rotated_gradient_op.cu
Fix unused variable warnings (#72410)
2022-02-07 18:25:29 +00:00
roi_align_rotated_gradient_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
roi_align_rotated_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
roi_align_rotated_op.cu
Drop more unused variables (#71123)
2022-01-11 12:46:24 -08:00
roi_align_rotated_op.h
remove default implementation for RoIAlignRotatedOp::RunOnDevice (#92885)
2023-01-24 07:20:37 +00:00
roi_pool_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
roi_pool_op.cu
Drop unused variables; make things const; use some auto (#71107)
2022-01-11 08:55:54 -08:00
roi_pool_op.h
refactor caffe2 operator constructors - 7/9 (#17088)
2019-02-28 14:23:53 -08:00
rowmul_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
rowmul_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
rsqrt_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
rsqrt_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
rsqrt_op.h
Add Cube and Cbrt Ops (#8991)
2018-06-28 14:55:30 -07:00
scale_blobs_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
scale_blobs_op.cu
Check all CUDA API calls for errors in caffe2/ (#81816)
2022-10-28 00:41:06 +00:00
scale_blobs_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
scale_op_gpu.cc
codemod: caffe::float16 -> at::Half (#11785)
2018-09-20 18:55:19 -07:00
scale_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
scale_op.h
Histogram Binning Calibration
2020-09-06 17:11:16 -07:00
segment_reduction_op_gpu.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
segment_reduction_op_gpu.cuh
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
segment_reduction_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
segment_reduction_op.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
self_binning_histogram_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
self_binning_histogram_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
selu_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
selu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
selu_op.h
refactor caffe2 operator constructors - 7/9 (#17088)
2019-02-28 14:23:53 -08:00
sequence_ops.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
sequence_ops.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
sequence_ops.h
refactor caffe2 operator constructors - 7/9 (#17088)
2019-02-28 14:23:53 -08:00
shape_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
shape_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
shape_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
sigmoid_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sigmoid_op_cudnn.cc
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
sigmoid_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
sigmoid_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
sigmoid_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
sin_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
sin_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
sin_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
sinh_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
sinh_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
sinh_op.h
Implement Sinh and Cosh (#9213)
2018-07-10 18:55:31 -07:00
sinusoid_position_encoding_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sinusoid_position_encoding_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
slice_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
slice_op.cu
Drop unused variables and add some const (#71106)
2022-01-11 12:38:59 -08:00
slice_op.h
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
softmax_op_cudnn.cc
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
softmax_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
softmax_op.h
Optimize SoftmaxOp on CPU (#18635)
2019-04-10 18:52:15 -07:00
softmax_ops.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
softmax_utils.cc
Optimize SoftmaxOp on CPU (#18635)
2019-04-10 18:52:15 -07:00
softmax_utils.h
Optimize SoftmaxOp on CPU (#18635)
2019-04-10 18:52:15 -07:00
softmax_with_loss_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
softmax_with_loss_op.h
Linearizable Label: Class Weights, Allow Missing Label, and Average by Batch Size (#29707)
2019-11-13 16:52:27 -08:00
softplus_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
softplus_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
softplus_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
softsign_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
softsign_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
softsign_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
space_batch_op_gpu.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
space_batch_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
space_batch_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
sparse_dropout_with_replacement_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
sparse_dropout_with_replacement_op.h
Implement dropout with replacement for id list features. (#22880)
2019-07-23 14:34:21 -07:00
sparse_itemwise_dropout_with_replacement_op.cc
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
sparse_itemwise_dropout_with_replacement_op.h
[itemwise-dropout][1/x][low-level module] Implement Itemwise Sparse Feature Dropout in Dper3 (#59322)
2021-06-04 19:59:17 -07:00
sparse_lp_regularizer_op_gpu.cu
Adding sparse Lp regularization operator to Caffe2 (#38574)
2020-06-01 15:21:19 -07:00
sparse_lp_regularizer_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sparse_lp_regularizer_op.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
sparse_normalize_op_gpu.cu
Windows DLL build with Caffe2 code (#11266)
2018-09-06 15:12:20 -07:00
sparse_normalize_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sparse_normalize_op.h
Renaming CAFFE2_API to TORCH_API (#49496)
2020-12-18 10:54:50 -08:00
sparse_to_dense_mask_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sparse_to_dense_mask_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
sparse_to_dense_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
sparse_to_dense_op.cu
use explicitly non-returning GPU atomics (#60607)
2021-06-28 18:17:29 -07:00
sparse_to_dense_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
spatial_batch_norm_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
spatial_batch_norm_op_cudnn.cu
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
spatial_batch_norm_op_impl.cuh
Drop unused variables; make things const; use some auto (#71107)
2022-01-11 08:55:54 -08:00
spatial_batch_norm_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
spatial_batch_norm_op.cu
Optimize SpatialBNOp on GPU (#16395)
2019-01-28 09:36:45 -08:00
spatial_batch_norm_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
spatial_softmax_with_loss_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
spatial_softmax_with_loss_op.h
refactor caffe2 operator constructors - 8/9 (#17089)
2019-02-28 14:45:20 -08:00
sqr_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
sqr_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
sqr_op.h
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
sqrt_op_gpu.cc
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
sqrt_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
sqrt_op.h
Seperate level1 elementwise functions from math (#16397)
2019-01-30 00:04:12 -08:00
square_root_divide_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
square_root_divide_op.h
[codemod] Fix shadows in PyTorch (#117562)
2024-01-17 20:33:50 +00:00
stats_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
stats_put_ops.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
stats_put_ops.h
Fix typos under caffe2/operators directory (#98235)
2023-04-05 06:26:01 +00:00
stop_gradient_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
stop_gradient.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
stop_gradient.h
Remove Context dependency from Tensor class (#14269)
2018-11-28 15:45:38 -08:00
string_ops_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
string_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
string_ops.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
stump_func_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
stump_func_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
stump_func_op.h
refactor caffe2 operator constructors - 8/9 (#17089)
2019-02-28 14:45:20 -08:00
stylizer_ops.cc
fix torch.config can't respect USE_MKLDNN flag issue (#75001)
2022-07-17 15:00:48 +00:00
summarize_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
summarize_op.cu
Replace all CHECK_ and DCHECK_ with TORCH_* macros (#82032)
2022-07-26 01:20:44 +00:00
summarize_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
swish_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
swish_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
swish_op.h
[Caffe2] Update elementwise ops to support numpy style boradcast (#8070)
2018-06-05 15:49:16 -07:00
tan_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
tan_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
tan_op.h
Update from Facebook (#8887)
2018-06-26 14:55:48 -07:00
tanh_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
tanh_op_cudnn.cc
Add cudnn activation ops (#9379)
2018-07-12 23:18:56 -07:00
tanh_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
tanh_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
tanh_op.h
Move tanh function to math (#9328)
2018-07-11 13:59:50 -07:00
tensor_protos_db_input_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
tensor_protos_db_input.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
tensor_protos_db_input.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
text_file_reader_utils_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
text_file_reader_utils.cc
[clang-tidy] Exclude cppcoreguidelines-avoid-magic-numbers (#57841)
2021-05-07 20:02:33 -07:00
text_file_reader_utils.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
text_file_reader.cc
Compile without -Wno-unused-variable (take 2) (#66041)
2021-10-04 20:39:39 -07:00
thresholded_relu_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
thresholded_relu_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
thresholded_relu_op.h
refactor caffe2 operator constructors - 8/9 (#17089)
2019-02-28 14:45:20 -08:00
tile_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
tile_op.cu
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
tile_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
top_k_heap_selection.cuh
[ROCm] Changes not to rely on CUDA_VERSION or HIP_VERSION (#65610)
2021-09-29 09:55:43 -07:00
top_k_radix_selection.cuh
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
top_k.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
top_k.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
top_k.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
transpose_op_cudnn.cc
[cuDNN] Cleanup cuDNN < 8.1 ifdefs (#120862)
2024-03-07 01:46:25 +00:00
transpose_op.cc
Change caffe2 branch links to main (#100129)
2023-04-27 10:31:50 +00:00
transpose_op.cu
Update math::Transpose to support tensor with size > 2G (#17670)
2019-03-20 18:22:21 -07:00
transpose_op.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
tt_linear_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
tt_linear_op.h
[BE] Enforce missing override keyword (#104032)
2023-06-24 02:34:24 +00:00
unique_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
unique_ops.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
unique_ops.h
Tensor reinitialization codemod - 4/5 (#15967)
2019-01-11 16:41:19 -08:00
unsafe_coalesce.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
unsafe_coalesce.cu
[C2] Revive unsafe CoalesceOp (#49402)
2020-12-17 04:31:29 -08:00
unsafe_coalesce.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
upsample_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
upsample_op.cu
Fix unused variable warnings (#72410)
2022-02-07 18:25:29 +00:00
upsample_op.h
refactor caffe2 operator constructors - 8/9 (#17089)
2019-02-28 14:45:20 -08:00
utility_ops_gpu_test.cc
Rename ndim() -> dim() - 5/6
2018-11-06 16:38:35 -08:00
utility_ops_test.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
utility_ops.cc
Revert "Use missing-prototypes in torch_cpu (#103725)"
2023-06-22 18:30:31 +00:00
utility_ops.cu
Use CUDA DSA in caffe2/operators (#95299)
2023-06-14 06:58:34 +00:00
utility_ops.h
[caffe2] Add enforce inside ScatterAssignOp (#106882)
2023-08-10 21:46:13 +00:00
variable_length_sequence_padding.cc
Fix some typos.
2022-04-11 21:55:59 +00:00
variable_length_sequence_padding.h
use irange for loops (#70248)
2022-01-06 23:14:29 -08:00
weighted_multi_sampling_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
weighted_multi_sampling_op.h
refactor caffe2 operator constructors - 9/9 (#17090)
2019-02-28 09:53:18 -08:00
weighted_sample_op.cc
Compile without -Wno-unused-variable (take 2) (#66041)
2021-10-04 20:39:39 -07:00
weighted_sample_op.cu
Check kernel launches in caffe2/operators (#52240)
2021-02-16 16:42:05 -08:00
weighted_sample_op.h
refactor caffe2 operator constructors - 9/9 (#17090)
2019-02-28 09:53:18 -08:00
while_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
while_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
while_op.h
refactor caffe2 operator constructors - 9/9 (#17090)
2019-02-28 09:53:18 -08:00
workspace_ops.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
zero_gradient_op_gpu.cc
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00
zero_gradient_op.cc
Disable avoid-non-const-global-variables lint check (#62008)
2021-07-22 18:04:40 -07:00
zero_gradient_op.h
Remove Apache headers from source.
2018-03-27 13:10:18 -07:00