OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
d0ad848aa5
pytorch/caffe2/operators
History
Howard Cheng dfb1b95919 [caffe2] Add enforce inside ScatterAssignOp (#106882) 
Summary: Adding an enforce gives better error information than raising SIGFPE when division by zero happens. We'll get the actual BlobRef names as well as the error categories.

Test Plan:
Ran a local worker and client using DPP session with empty tensors and checked the error:

`../buck-out/v2/gen/fbcode/data_preproc/perf_test/client --sr2_event_base_pool_size=24`

`../buck-out/v2/gen/fbcode/data_preproc/perf_test/worker --dpp_session_id=5D49F56C98CC95BD97027BC0DDB38D8F`

```{dpp_internal_errorcategory : user_error,
ONCALL : MLDP_CONTROL,
CATEGORY : INPUT_ERROR,
errorsubsystemtags : [DPP_WORKER],
errorcause : USER_ERROR,
RETRYABILITY : 0}F0806 17:47:52.607200 2280375 SchedRuntimeEnv.cpp:385] facebook::data_preproc::NonRetryableGenericUser
Error: User preprocessing error c10::Error: [enforce fail at utility_ops.h:730] input.numel() > 0. 0 vs 0. tensor has t
o be nonempty (Error from operator:
input: "preproc_data_pipeline/preproc/features/default_feature_preproc/normalization/dper_feature_normalization/sparse_
features_processor_1/sparse_feature_transform/F3_ADFINDER_USER_ADS_COFFEE_LSF_FLEXIBLE_BATCH_USER_FB_UIP_FEATURE_IDSCOR
ELIST_ENCODED_FB_UIP_TOP100_IDSCORELIST_ENCODED_1/sequential_1019/id_score_list_quantization_decode_1/Concat:0" input:
"preproc_data_pipeline/preproc/features/default_feature_preproc/normalization/dper_feature_normalization/sparse_feature
s_processor_1/sparse_feature_transform/F3_ADFINDER_USER_ADS_COFFEE_LSF_FLEXIBLE_BATCH_USER_FB_UIP_FEATURE_IDSCORELIST_E
NCODED_FB_UIP_TOP100_IDSCORELIST_ENCODED_1/sequential_1019/id_score_list_quantization_decode_1/Mul_2" input: "preproc_d
ata_pipeline/preproc/features/default_feature_preproc/normalization/dper_feature_normalization/sparse_features_processo
r_1/sparse_feature_transform/F3_ADFINDER_USER_ADS_COFFEE_LSF_FLEXIBLE_BATCH_USER_FB_UIP_FEATURE_IDSCORELIST_ENCODED_FB_UIP_TOP100_IDSCORELIST_ENCODED_1/sequential_1019/id_score_list_quantization_decode_1/encoded_id_lengths" output: "preproc_data_pipeline/preproc/features/default_feature_preproc/normalization/dper_feature_normalization/sparse_features_processor_1/sparse_feature_transform/F3_ADFINDER_USER_ADS_COFFEE_LSF_FLEXIBLE_BATCH_USER_FB_UIP_FEATURE_IDSCORELIST_ENCODED_FB_UIP_TOP100_IDSCORELIST```

Differential Revision: D48104430

Pull Request resolved: https://github.com/pytorch/pytorch/pull/106882
Approved by: https://github.com/kit1980
2023-08-10 21:46:13 +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 [BE] Enforce missing override keyword (#104032) 2023-06-24 02:34:24 +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
accumulate_op.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
accumulate_op.cu
accumulate_op.h
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
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
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
assert_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
assert_op.cu
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
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
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
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
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
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
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
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
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
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
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
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
conv_op_cache_cudnn.cc
conv_op_cache_cudnn.h
conv_op_cudnn.cc Compile without -Wno-unused-variable (take 2) (#66041) 2021-10-04 20:39:39 -07: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
conv_op_impl.h use irange for loops 6 (#66742) 2021-12-07 16:07:50 -08:00
conv_op_shared_gpu.cc
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
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 Remove deprecated cuDNN API from caffe2 (#38680) 2020-05-20 12:55:58 -07:00
conv_transpose_op_gpu.cc
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
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
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
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
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
counter_ops_gpu.cc
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
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
data_couple_gpu.cu
data_couple.cc Fix typos under caffe2/operators directory (#98235) 2023-04-05 06:26:01 +00:00
data_couple.h
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
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
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
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
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] use hipblas instead of rocblas (#105881) 2023-07-31 20:42:55 +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
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
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
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
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
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
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
exp_op_gpu.cc
exp_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
exp_op.h
expand_op_gpu.cc
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
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
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
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
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
free_op.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
free_op.h
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 Fix typos under caffe2/operators directory (#98235) 2023-04-05 06:26:01 +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 Fix typos under caffe2/operators directory (#98235) 2023-04-05 06:26:01 +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
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
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
if_op.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
if_op.h
im2col_op_gpu.cc
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
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
jsd_op.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
jsd_op.h
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
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
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
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
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
locally_connected_op_impl.h use irange for loops 6 (#66742) 2021-12-07 16:07:50 -08:00
locally_connected_op_util.cc
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
log_op_gpu.cc
log_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
log_op.h
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
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
loss_op.h
lp_pool_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +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
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
matmul_op_gpu.cc
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
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
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
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
negate_gradient_op_gpu.cc
negate_gradient_op.cc Fix typos under caffe2/operators directory (#98235) 2023-04-05 06:26:01 +00:00
negate_gradient_op.h
negative_op_gpu.cc
negative_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
negative_op.h
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
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 use irange for loops 6 (#66742) 2021-12-07 16:07:50 -08:00
operator_fallback_gpu_test.cc
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
order_switch_ops_gpu.cc
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
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
pool_op_util.cc Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
pool_op_util.h
pool_op.cc Revert "Use missing-prototypes in torch_cpu (#103725)" 2023-06-22 18:30:31 +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
prepend_dim_op_gpu.cc
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
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
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
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
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
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
reshape_op_gpu.cc
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
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
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
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
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
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
shape_op_gpu.cc
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
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
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
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
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
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
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
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 [caffe2] Use extended versions of cuDNN calls for SpatialBN 2021-03-05 18:18:15 -08: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
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
sqr_op_gpu.cc
sqr_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
sqr_op.h
sqrt_op_gpu.cc
sqrt_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
sqrt_op.h
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 use irange for loops (#70248) 2022-01-06 23:14:29 -08: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
stop_gradient.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
stop_gradient.h
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
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
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
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
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
tensor_protos_db_input_gpu.cc
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
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 Disable cudnn transpose for int types (#26934) 2019-09-27 11:36:10 -07:00
transpose_op.cc Change caffe2 branch links to main (#100129) 2023-04-27 10:31:50 +00:00
transpose_op.cu
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
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
utility_ops_gpu_test.cc
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
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
while_op_gpu.cc
while_op.cc Disable avoid-non-const-global-variables lint check (#62008) 2021-07-22 18:04:40 -07:00
while_op.h
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
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