OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 00:21:07 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
71cb13869b
pytorch/caffe2/operators
History
Dianshi Li 31f605344f [Resubmit][S372460 follow up] Reduce embedding feature validation failure carry-on impact (#111838) 
Summary:
## Context
The embedding feature validation for GatherRangeToDense was added in the previous diff: D18031155. The logic will check the mismatchedranges or empty ranges in the whole model lifecycle, once the ratio exceeds some thresehold, it will trigger ENFORCE failure (exception).
In the current implementation, it may have carry-on impact. The mismatch ratio is equal to:
```
ratio = mismatched_ranges_from_t0_to_t1 / total_ranges_from_t0_to_t1
```
if the mismatched_ranges_from_t0 somehow increased a lot (bad value spike) at request N, the ratio will be much larger than the treshold.  Then it may take long util t2 to make the new ratio drops below the threshold. however, the requests between t1 and t2 may be all good requests, then it brings carry-on impact.
Instead, we would like to propose a new strategy, when exception happen at T1, we will clean up all the history counters for this bad feature, and make it a clean run for the next phase util the next exception. it then will get rid of the carry-on impact.
In this logic, we will clean up the counter based on the bad feature J.
more context: https://docs.google.com/document/d/1tYHISyiLf-PVKPVGlZRZ0iq2Hvog3g5BjCKMCDLZHHo/edit

Test Plan:
hardcode a much smaller threshold as 0.0001 to force trigger the exception, deploy on some hosts in prod tiers

```
EPHEMERAL_PACKAGE=d44a3de1305c3b4c30fd62bc354a1285 tw update fbcode/tupperware/config/admarket/sigrid/predictor/prod.tw tsp_cln/admarket/sigrid_predictor_v2_dh_t1_elastic_ha --tasks=100-199 --fast --force
```

## 1) totalRanges can be correct logged
```
I1018 20:37:09.012523  2074 gather_ranges_to_dense_op.h:69 req:00f00000001abbf1] In GatherRangesToDenseOp:
  Lifetime empty ranges for each feature is 12354.
  Lifetime mismatched ranges for each feature is 526.
  With a total of 87503 examples for each feature.
```

## 2) exception can be still triggered
```
E1018 21:08:42.007398   668 LoggingPredictorService.cpp:701 req:001000000013df51] getRequestPrecomputedDataOnePass failure on model 481948521_146: [enforce fail at gather_ranges_to_dense_op.h:215] std::max(totalRangesTemp, minObservation_) * maxMismatchedRatio_ >= mismatchedRangesTemp. 0.1 vs 1. Ratio of range length mismatch for feature at index 0 is 0.00813008 (1/123) which exceeds 1e-05. The incorrect lengths include: 15 (Error from operator:

```

Differential Revision: D50570811

Pull Request resolved: https://github.com/pytorch/pytorch/pull/111838
Approved by: https://github.com/malfet
2023-10-28 03:50:33 +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 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
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
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 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
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
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 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
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 Rename IntList to IntArrayRef. (#16751) 2019-02-05 14:54:34 -08:00
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 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
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 refactor caffe2 operator constructors - 2/9 (#17083) 2019-02-28 14:23:55 -08:00
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 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
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 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
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 refactor caffe2 operator constructors - 3/9 (#17084) 2019-02-28 14:13:17 -08: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
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 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
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 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
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
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 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
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 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 refactor caffe2 operator constructors - 4/9 (#17085) 2019-02-28 14:23:52 -08:00
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 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
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 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
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 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
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 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
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 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
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 Change ConvPoolOp<Context>::SetOutputSize to ConvPoolOp<Context>::GetOutputSize (#17764) 2019-03-07 18:38:53 -08: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
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
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 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
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 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
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 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
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 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
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 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 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 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
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 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 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 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
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
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
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