Scott Wolchok 1e0809dbf9 [PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/50385 We no longer use this flag internally, and it's not referenced externally either, so let's clean up. ghstack-source-id: 119676743 Test Plan: CI Reviewed By: ezyang Differential Revision: D25852220 fbshipit-source-id: a4427edff6cbb241340f9f6ae6db4e74832949c2		2021-01-20 10:26:54 -08:00
..
hip	[ROCm][Caffe2] Enable MIOpen 3D Pooling (#38260)	2020-07-08 17:42:55 -07:00
quantized	Adjust bound_shape_inferencer to take 4 inputs for FCs (#41934)	2020-08-05 18:44:48 -07:00
rnn	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
abs_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
abs_op.cu
abs_op.h
accumulate_op.cc
accumulate_op.cu
accumulate_op.h
accuracy_op.cc
accuracy_op.cu
accuracy_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
acos_op.cc
acos_op.cu
acos_op.h
activation_ops_cudnn.h
affine_channel_op.cc
affine_channel_op.cu
affine_channel_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
alias_with_name.cc
alias_with_name.cu
alias_with_name.h
apmeter_op.cc
apmeter_op.h
arg_ops.cc
arg_ops.cu
arg_ops.h
asin_op.cc
asin_op.cu
asin_op.h
assert_op.cc
assert_op.cu
assert_op.h
async_net_barrier_op.cc	[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.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	[DPER] Introduce barrier operation to force synchronization of threads in async execution (#49322)	2020-12-15 16:13:42 -08:00
atan_op.cc
atan_op.cu
atan_op.h
atomic_ops.cc
batch_box_cox_op.cc	Export box_cox operator in caffe2	2020-06-17 19:28:53 -07:00
batch_box_cox_op.h	Export box_cox operator in caffe2	2020-06-17 19:28:53 -07:00
batch_bucketize_op.cc
batch_bucketize_op.h
batch_gather_ops.cc
batch_gather_ops.cu
batch_gather_ops.h
batch_matmul_op.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
batch_matmul_op.cu	RIP CUDA <9.2: circleci, aten, and caffe2 (#36846)	2020-05-18 13:41:05 -07:00
batch_matmul_op.h
batch_matmul_op_gpu_test.cc
batch_matmul_op_test.cc
batch_moments_op.cc
batch_moments_op.cu
batch_moments_op.h
batch_permutation_op.cc	Add zero input support for batch permutation op (#39851)	2020-06-13 21:34:24 -07:00
batch_permutation_op.cu	Add zero input support for batch permutation op (#39851)	2020-06-13 21:34:24 -07:00
batch_permutation_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_sparse_to_dense_op.cc
batch_sparse_to_dense_op.cu
batch_sparse_to_dense_op.h
bbox_transform_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
bbox_transform_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
bisect_percentile_op.cc
bisect_percentile_op.h
boolean_mask_ops.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
boolean_mask_ops.cu
boolean_mask_ops.h
boolean_unmask_ops.cc
boolean_unmask_ops.cu
boolean_unmask_ops.h
boolean_unmask_ops_test.cc
box_with_nms_limit_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
box_with_nms_limit_op.h	BoxWithNMSLimit support int batch_splits input (#47504)	2020-11-07 00:27:51 -08:00
bucketize_op.cc
bucketize_op.cu	Prevent overflow in bucketize binary search	2020-09-30 15:04:11 -07: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
byte_weight_dequant_op.h	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
cast_op.cc
cast_op.cu
cast_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
cbrt_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
cbrt_op.cu
cbrt_op.h
cc_bmm_bg_op.cc
cc_bmm_bg_op.h
ceil_op.cc
ceil_op.cu
ceil_op.h
channel_backprop_stats_op.cc
channel_backprop_stats_op.cu
channel_backprop_stats_op.h
channel_shuffle_op.cc
channel_shuffle_op.cu	Expose ChannelShuffle (#46000)	2020-10-08 16:00:01 -07:00
channel_shuffle_op.h
channel_stats_op.cc
channel_stats_op.cu
channel_stats_op.h
channelwise_conv3d_op_cudnn.cu
clip_op.cc
clip_op.cu
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
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.cc
communicator_op_gpu.cc
concat_split_op.cc	Fix illegal memory acess issue for CUDA versionn of SplitByLengths operator.	2020-08-14 01:04:08 -07:00
concat_split_op.h	Fix illegal memory acess issue for CUDA versionn of SplitByLengths operator.	2020-08-14 01:04:08 -07:00
concat_split_op_gpu.cc
conditional_op.cc
conditional_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
conv_gradient_op.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
conv_op.cc
conv_op.h
conv_op_cache_cudnn.cc
conv_op_cache_cudnn.h
conv_op_cache_cudnn_test.cc
conv_op_cudnn.cc	Remove deprecated cuDNN API from caffe2 (#38680)	2020-05-20 12:55:58 -07:00
conv_op_eigen.cc
conv_op_gpu.cc
conv_op_impl.h
conv_op_shared.cc
conv_op_shared.h
conv_op_shared_gpu.cc
conv_pool_op_base.h	Fix signed-unsigned warnings (RELAND) (#36224)	2020-04-08 16:29:27 -07:00
conv_transpose_gradient_op.cc
conv_transpose_op.cc
conv_transpose_op.h
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
conv_transpose_op_mobile.cc
conv_transpose_op_mobile.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
conv_transpose_op_mobile_impl.h
conv_transpose_op_mobile_test.cc	[PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385)	2021-01-20 10:26:54 -08:00
conv_transpose_unpool_op_base.h
copy_op.cc	To fix caffe2 model with Copy OP cannot export to onnx model (#37144)	2020-05-04 11:34:09 -07:00
copy_op.cu
copy_op.h
copy_rows_to_tensor_op.cc
copy_rows_to_tensor_op.h
cos_op.cc
cos_op.cu
cos_op.h
cosh_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
cosh_op.cu
cosh_op.h
cosine_embedding_criterion_op.cc
cosine_embedding_criterion_op.cu
cosine_embedding_criterion_op.h
counter_ops.cc
counter_ops.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
counter_ops_gpu.cc
crash_op.cc
create_scope_op.cc
create_scope_op.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
crf_viterbi_op.cc
cross_entropy_op.cc	[Format] format a few files (#35187)	2020-04-17 14:30:01 -07:00
cross_entropy_op.cu
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
ctc_beam_search_decoder_op.h
ctc_greedy_decoder_op.cc
ctc_greedy_decoder_op.h
cube_op.cc
cube_op.cu
cube_op.h
data_couple.cc
data_couple.h
data_couple_gpu.cu
dataset_ops.cc	pass TypeMeta by value (#45026)	2020-10-30 10:14:17 -07:00
dataset_ops.h	pass TypeMeta by value (#45026)	2020-10-30 10:14:17 -07:00
deform_conv_gradient_op.cc
deform_conv_op.cc
deform_conv_op.cu
deform_conv_op.h
deform_conv_op_impl.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
dense_vector_to_id_list_op.cc
dense_vector_to_id_list_op.h
depthwise_3x3_conv_op_cudnn.cu
distance_op.cc
distance_op.cu
distance_op.h
do_op.cc
do_op.h
do_op_gpu.cc
dropout_op.cc	caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987)	2020-10-16 16:08:35 -07:00
dropout_op.cu
dropout_op.h
dropout_op_cudnn.cc
elementwise_add_gradient_op.cc
elementwise_add_op.cc
elementwise_add_op.h
elementwise_add_op_gpu.cc
elementwise_div_gradient_op.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
elementwise_div_op.cc
elementwise_div_op.cu
elementwise_div_op.h
elementwise_linear_op.cc
elementwise_linear_op.cu
elementwise_linear_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
elementwise_logical_ops.cc
elementwise_logical_ops.h
elementwise_mul_gradient_op.cc	Fix compilation errors for enabling Intel nextgen compiler (icx/icpx) (#35939)	2020-04-07 09:31:32 -07:00
elementwise_mul_op.cc
elementwise_mul_op.cu	[C2] Optimize MulGradient Operator when inner_size is 1 (#36767)	2020-04-24 11:17:49 -07:00
elementwise_mul_op.h
elementwise_op_gpu_test.cc
elementwise_op_test.cc
elementwise_op_test.h
elementwise_ops.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
elementwise_ops.cu	[ROCm] [ROCm 2.10+] enable fp16 dot in Caffe2 backend (#30432)	2020-04-03 08:01:23 -07:00
elementwise_ops.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
elementwise_ops_schema.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
elementwise_ops_utils.cc	[Caffe2]Add more error message in ComputeBinaryBroadcastForwardDims	2020-12-07 07:42:49 -08:00
elementwise_ops_utils.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
elementwise_sub_gradient_op.cc
elementwise_sub_op.cc
elementwise_sub_op.h
elementwise_sub_op_gpu.cc
elementwise_sum_op.cc
elu_op.cc
elu_op.cu
elu_op.h
elu_op_cudnn.cc
enforce_finite_op.cc
enforce_finite_op.cu
enforce_finite_op.h
ensure_clipped_op.cc
ensure_clipped_op.h
ensure_cpu_output_op.cc
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
erf_op.cu
erf_op.h
exp_op.cc
exp_op.h
exp_op_gpu.cc
expand_op.cc
expand_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
expand_op_gpu.cc
expand_squeeze_dims_op.cc
expand_squeeze_dims_op.h
expand_squeeze_dims_op_gpu.cc
fc_inference.cc	[caffe2] add cost inference for FusedFakeQuantFC and FusedFakeQuantFCGradient (#44840)	2020-09-17 14:07:17 -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	Add a new op for converting the dense feature to sparse representation	2020-07-27 12:45:37 -07:00
feature_maps_ops.h	Add a new op for converting the dense feature to sparse representation	2020-07-27 12:45:37 -07:00
feed_blob_op.cc
feed_blob_op.h
filler_op.cc	topk tensor k support (#39407)	2020-06-15 13:10:20 -07:00
filler_op.cu
filler_op.h	caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987)	2020-10-16 16:08:35 -07:00
find_duplicate_elements_op.cc
find_duplicate_elements_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
find_op.cc
find_op.cu
find_op.h
flatten_op.cc
flatten_op.h
flexible_top_k.cc
flexible_top_k.h
floor_op.cc
floor_op.cu
floor_op.h
free_op.cc
free_op.h
free_op_gpu.cc
fully_connected_op.cc	[caffe2] add cost inference for FusedFakeQuantFC and FusedFakeQuantFCGradient (#44840)	2020-09-17 14:07:17 -07:00
fully_connected_op.h	[Format] format a few files (#35187)	2020-04-17 14:30:01 -07:00
fully_connected_op_gpu.cc	RIP CUDA <9.2: circleci, aten, and caffe2 (#36846)	2020-05-18 13:41:05 -07:00
fused_rowwise_8bit_conversion_ops.cc
fused_rowwise_8bit_conversion_ops.h	[caffe2] optimize 2/4-bit row-wise quantization (#387)	2020-06-19 21:28:31 -07:00
fused_rowwise_nbit_conversion_ops.cc	fp16 include not needed (#35708)	2020-03-30 17:47:44 -07:00
fused_rowwise_nbit_conversion_ops.h	[caffe2] optimize 2/4-bit row-wise quantization (#387)	2020-06-19 21:28:31 -07:00
fused_rowwise_nbitfake_conversion_ops.cc	fp16 include not needed (#35708)	2020-03-30 17:47:44 -07:00
fused_rowwise_nbitfake_conversion_ops.h	[caffe2] minor typo fix in fused_rowwise_nbitfake_conversion_ops.h comment (#39315)	2020-05-31 23:32:39 -07:00
fused_rowwise_random_quantization_ops.cc
fused_rowwise_random_quantization_ops.h
gather_fused_8bit_rowwise_op.cc
gather_fused_8bit_rowwise_op.h
gather_op.cc
gather_op.cu
gather_op.cuh
gather_op.h
gather_ranges_to_dense_op.cc	Resend diff D23858329 (#45315)	2020-09-24 18:41:49 -07:00
gather_ranges_to_dense_op.h	Replace GatherRangesToDense operator in Dper from c2 to pt.	2020-11-20 08:14:32 -08:00
gelu_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
gelu_op.cu
gelu_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
generate_proposals_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
generate_proposals_op.cu
generate_proposals_op.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
generate_proposals_op_gpu_test.cc
generate_proposals_op_test.cc
generate_proposals_op_util_boxes.h
generate_proposals_op_util_boxes_test.cc
generate_proposals_op_util_nms.h
generate_proposals_op_util_nms_gpu.cu
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_gpu_test.cc
generate_proposals_op_util_nms_test.cc
given_tensor_byte_string_to_uint8_fill_op.cc
given_tensor_byte_string_to_uint8_fill_op.cu
given_tensor_byte_string_to_uint8_fill_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
given_tensor_fill_op.cc
given_tensor_fill_op.cu
given_tensor_fill_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
glu_op.cc
glu_op.cu
glu_op.h
group_norm_op.cc
group_norm_op.cu
group_norm_op.h
gru_unit_op.cc
gru_unit_op.h
gru_unit_op_gpu.cu
h_softmax_op.cc
h_softmax_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
half_float_ops.cc	Enable fp16 for UniformFill (#44540)	2020-09-15 15:09:18 -07:00
half_float_ops.cu	Enable fp16 for UniformFill (#44540)	2020-09-15 15:09:18 -07:00
half_float_ops.h	Enable fp16 for UniformFill (#44540)	2020-09-15 15:09:18 -07:00
half_float_ops_test.cc
hard_sigmoid_op.cc
hard_sigmoid_op.cu
hard_sigmoid_op.h
heatmap_max_keypoint_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
heatmap_max_keypoint_op.h
histogram_op.cc	[dper][pruning] add histogram op (#38514)	2020-05-28 15:45:04 -07:00
histogram_op.h	[dper][pruning] add histogram op (#38514)	2020-05-28 15:45:04 -07:00
if_op.cc
if_op.h
if_op_gpu.cc
im2col_op.cc
im2col_op.h
im2col_op_gpu.cc
index_hash_ops.cc
index_hash_ops.h	[c10/cuda] Reorganize device_count() and robustly surface ASAN warnings (#42249)	2020-08-05 11:39:31 -07:00
index_ops.cc
index_ops.h	pass TypeMeta by value (#45026)	2020-10-30 10:14:17 -07:00
inference_lstm_op.cc
inference_lstm_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
instance_norm_gradient_op.cc
instance_norm_op.cc
instance_norm_op.cu
instance_norm_op.h
integral_image_op.cc
integral_image_op.cu
integral_image_op.h
is_empty_op.cc
is_empty_op.h
jsd_op.cc
jsd_op.h
key_split_ops.cc
key_split_ops.h
last_n_window_collector.cc	[mlf][efficiency] add tensor inference function to last-n collector op (#46693)	2020-10-22 01:15:00 -07:00
layer_norm_op.cc
layer_norm_op.cu
layer_norm_op.h	[Caffe2] Fix LayerNormOp when batch_size == 0. (#45250)	2020-09-24 12:30:03 -07:00
leaky_relu_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
leaky_relu_op.cu
leaky_relu_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
length_split_op.cc
length_split_op.h
lengths_pad_op.cc
lengths_pad_op.cu
lengths_pad_op.h
lengths_reducer_fused_8bit_rowwise_ops.cc	[caffe] fix input order in SLS op documentation (#36708)	2020-04-16 00:55:54 -07:00
lengths_reducer_fused_8bit_rowwise_ops.h	[caffe2] explicitly pass use_offsets=false when calling fbgemm embedding kernels (#35711)	2020-03-31 08:35:19 -07:00
lengths_reducer_fused_nbit_rowwise_ops.cc	Expose SparseLengthsSum8BitRowwiseSparse to C10 (#47306)	2020-11-03 22:51:12 -08:00
lengths_reducer_fused_nbit_rowwise_ops.h	Expose SparseLengthsSum8BitRowwiseSparse to C10 (#47306)	2020-11-03 22:51:12 -08:00
lengths_reducer_ops.cc	Back out "Revert D19987020: [pytorch][PR] Add the sls tensor train op" (#43938)	2020-09-01 11:42:12 -07:00
lengths_reducer_ops.h	Back out "Revert D19987020: [pytorch][PR] Add the sls tensor train op" (#43938)	2020-09-01 11:42:12 -07:00
lengths_reducer_rowwise_8bit_ops.cc
lengths_reducer_rowwise_8bit_ops.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
lengths_tile_op.cc
lengths_tile_op.cu
lengths_tile_op.h
lengths_top_k_op.cc
lengths_top_k_op.h
listwise_l2r_op.cc
listwise_l2r_op.h
load_save_op.cc
load_save_op.h
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
local_response_normalization_op.cc
local_response_normalization_op.cu
local_response_normalization_op.h
local_response_normalization_op_cudnn.cc
locally_connected_op.cc
locally_connected_op.h
locally_connected_op_gpu.cc
locally_connected_op_impl.h
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
log_op.cc
log_op.h
log_op_gpu.cc
logit_op.cc	Export logic op to pytorch	2020-07-08 02:27:09 -07:00
logit_op.cu
logit_op.h	Export logic op to pytorch	2020-07-08 02:27:09 -07:00
loss_op.cc
loss_op.cu
loss_op.h
lp_pool_op.cc	Remove 4 unused variables in lp_pool_op.cc (#42329)	2020-07-30 15:50:17 -07:00
lp_pool_op.cu
lpnorm_op.cc	Back out "Back out "[c2] register cuda op for LpNorm (fallback)"" (#38566)	2020-05-19 10:37:25 -07: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.cc
lstm_unit_op.h	[caffe2] Explicit vectorization of LSTM operator (#35556)	2020-04-01 17:19:56 -07:00
lstm_unit_op_gpu.cu
lstm_utils.h
map_ops.cc
map_ops.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
margin_ranking_criterion_op.cc
margin_ranking_criterion_op.cu
margin_ranking_criterion_op.h
matmul_op.cc
matmul_op.h
matmul_op_gpu.cc
max_pool_with_index.cu
max_pool_with_index_gpu.h
mean_op.cc
mean_op.cu
mean_op.h	Adding Type Double to Caffe2 Mean Op	2020-09-28 13:35:29 -07:00
mem_query_op.cu
merge_id_lists_op.cc	Export MergeIdLists Caffe2 Operator to PyTorch	2020-08-14 14:46:17 -07:00
merge_id_lists_op.h	Export MergeIdLists Caffe2 Operator to PyTorch	2020-08-14 14:46:17 -07:00
minmax_gradient_ops.cc
minmax_ops.cc
minmax_ops.cu
minmax_ops.h
mish_op.cc	[Caffe2] Optimize MishOp on CPU (#48212)	2020-11-19 14:17:27 -08:00
mish_op.h	[Caffe2] Optimize MishOp on CPU (#48212)	2020-11-19 14:17:27 -08:00
mod_op.cc	[uhm][0/n] add cuda Mod Op (#46732)	2020-10-26 11:07:51 -07:00
mod_op.cu	[uhm][0/n] add cuda Mod Op (#46732)	2020-10-26 11:07:51 -07:00
mod_op.h
moments_op.cc
moments_op.cu
moments_op.h
multi_class_accuracy_op.cc
multi_class_accuracy_op.cu
multi_class_accuracy_op.h
negate_gradient_op.cc
negate_gradient_op.h
negate_gradient_op_gpu.cc
negative_op.cc
negative_op.h
negative_op_gpu.cc
ngram_ops.cc
ngram_ops.h
no_default_engine_op.h
norm_planar_yuv_op.cc
normalize_l1_op.cc
normalize_l1_op.h	Revert D22330340: [C2] Fixed a bug in normalization operator	2020-07-02 16:05:23 -07:00
normalize_op.cc
normalize_op.h	Revert D22330340: [C2] Fixed a bug in normalization operator	2020-07-02 16:05:23 -07:00
normalize_ops.cu
numpy_tile_op.cc
numpy_tile_op.h	pass TypeMeta by value (#45026)	2020-10-30 10:14:17 -07:00
one_hot_ops.cc	Export BatchBucketOneHot Caffe2 Operator to PyTorch	2020-08-11 14:00:19 -07:00
one_hot_ops.cu
one_hot_ops.h	Export BatchBucketOneHot Caffe2 Operator to PyTorch	2020-08-11 14:00:19 -07:00
onnx_while_op.cc
onnx_while_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
op_utils_cudnn.h
operator_fallback_gpu.h
operator_fallback_gpu_test.cc
order_switch_ops.cc
order_switch_ops.h
order_switch_ops_cudnn.cc
order_switch_ops_gpu.cc
pack_rnn_sequence_op.cc
pack_rnn_sequence_op.h
pack_segments.cc	[dper3] Export PackSegments and UnpackSegments to Pytorch	2020-09-11 09:29:24 -07:00
pack_segments.cu
pack_segments.h	[dper3] Export PackSegments and UnpackSegments to Pytorch	2020-09-11 09:29:24 -07:00
pad_op.cc
pad_op.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
pad_op_gpu.cu
partition_ops.cc
partition_ops.h
percentile_op.cc	Exposing Percentile Caffe2 Operator in PyTorch	2020-08-07 16:22:37 -07:00
percentile_op.h	Exposing Percentile Caffe2 Operator in PyTorch	2020-08-07 16:22:37 -07:00
perplexity_op.cc
perplexity_op.cu
perplexity_op.h
piecewise_linear_transform_op.cc
piecewise_linear_transform_op.cu
piecewise_linear_transform_op.h
pool_gradient_op.cc
pool_op.cc
pool_op.cu
pool_op.h
pool_op_cudnn.cc
pool_op_util.cc
pool_op_util.h
pow_op.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
pow_op.cu
pow_op.h	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
prefetch_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
prelu_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
prelu_op.cu
prelu_op.h
prepend_dim_op.cc
prepend_dim_op.h
prepend_dim_op_gpu.cc
quant_decode_op.cc
quant_decode_op.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
quantile_op.cc
quantile_op.h
rank_loss_op.cc
rank_loss_op.h
reciprocal_gradient_op.cc
reciprocal_op.cc
reciprocal_op.cu
reciprocal_op.h
reduce_front_back_max_ops.cc
reduce_front_back_max_ops.cu
reduce_front_back_max_ops.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
reduce_front_back_mean_ops.cc
reduce_front_back_sum_mean_ops.cu
reduce_front_back_sum_mean_ops.h	[codemod][caffe2] Run clang-format - 5/7	2020-06-30 15:45:11 -07:00
reduce_front_back_sum_ops.cc
reduce_ops.cc	Fix signed-unsigned warnings (#34791)	2020-03-19 00:29:56 -07:00
reduce_ops.cu
reduce_ops.h
reducer_functors.h
reduction_ops.cc
reduction_ops.cu
reduction_ops.h	Histogram Binning Calibration	2020-09-06 17:11:16 -07:00
relu_n_op.cc
relu_n_op.cu
relu_n_op.h
relu_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
relu_op.cu	Fix another case of float2::x and float2::y may not be the same on ROCm (#35785)	2020-04-01 14:14:30 -07:00
relu_op.h
remove_data_blocks_op.cc
remove_data_blocks_op.h
replace_nan_op.cc
replace_nan_op.cu
replace_nan_op.h
reservoir_sampling.cc	caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987)	2020-10-16 16:08:35 -07:00
reshape_op.cc
reshape_op.h
reshape_op_gpu.cc
reshape_op_gpu_test.cc
resize_3d_op.cc
resize_3d_op.cu
resize_3d_op.h
resize_op.cc
resize_op.cu
resize_op.h
reverse_packed_segs_op.cc
reverse_packed_segs_op.cu
reverse_packed_segs_op.h
rmac_regions_op.cc
rmac_regions_op.cu
rmac_regions_op.h
rms_norm_op.cc	[Caffe2] Add RMSNormOp (#44338)	2020-09-08 23:50:44 -07:00
rms_norm_op.cu	[Caffe2] Add RMSNormOp (#44338)	2020-09-08 23:50:44 -07:00
rms_norm_op.h	[Caffe2] Add RMSNormOp (#44338)	2020-09-08 23:50:44 -07:00
roi_align_gradient_op.cc	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_align_gradient_op.cu	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_align_gradient_op.h	Export roi_align_gradient_op to c10 (#34776)	2020-03-15 02:43:39 -07:00
roi_align_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
roi_align_op.cu	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_align_op.h	Refactor RoIAlignOp on CPU (#34698)	2020-03-27 07:53:58 -07:00
roi_align_op_gpu_test.cc	Re-enable Caffe2 test RoiAlignTest.CheckCPUGPUEqual (#40901)	2020-07-02 11:22:23 -07:00
roi_align_rotated_gradient_op.cc
roi_align_rotated_gradient_op.cu	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_align_rotated_gradient_op.h
roi_align_rotated_op.cc	remove ops in the __caffe2 namespace (#47318)	2020-11-16 15:30:16 -08:00
roi_align_rotated_op.cu	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_align_rotated_op.h
roi_pool_op.cc	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_pool_op.cu	add dllexport before template specialization functions for windows build (#45477)	2020-09-30 10:39:23 -07:00
roi_pool_op.h
rowmul_op.cc
rowmul_op.h
rsqrt_op.cc
rsqrt_op.cu
rsqrt_op.h
scale_blobs_op.cc
scale_blobs_op.cu
scale_blobs_op.h
scale_op.cc
scale_op.h	Histogram Binning Calibration	2020-09-06 17:11:16 -07:00
scale_op_gpu.cc
segment_reduction_op.cc
segment_reduction_op.h	TensorInferenceFunction checks	2020-10-11 16:08:58 -07:00
segment_reduction_op_gpu.cu	Enable fp16 for CUDA SparseLengthsSum/Mean (#44089)	2020-09-15 11:10:54 -07:00
segment_reduction_op_gpu.cuh	Enable fp16 for CUDA SparseLengthsSum/Mean (#44089)	2020-09-15 11:10:54 -07:00
self_binning_histogram_op.cc	[MLF] Allow for computing prune quantile thresholds on absolute value of indicators in distributed-inference-compatible embedding LUT pruning (#46789)	2020-11-02 11:31:31 -08:00
self_binning_histogram_op.h	[MLF] Allow for computing prune quantile thresholds on absolute value of indicators in distributed-inference-compatible embedding LUT pruning (#46789)	2020-11-02 11:31:31 -08:00
selu_op.cc
selu_op.cu
selu_op.h
sequence_ops.cc
sequence_ops.cu
sequence_ops.h
shape_op.cc
shape_op.h
shape_op_gpu.cc
sigmoid_gradient_op.cc
sigmoid_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
sigmoid_op.cu
sigmoid_op.h
sigmoid_op_cudnn.cc
sin_op.cc
sin_op.cu
sin_op.h
sinh_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
sinh_op.cu
sinh_op.h
sinusoid_position_encoding_op.cc
sinusoid_position_encoding_op.h
slice_op.cc	[caffe2] SliceOp axes indexing fixes. (#45432)	2020-10-06 13:21:08 -07:00
slice_op.cu	[caffe2] SliceOp axes indexing fixes. (#45432)	2020-10-06 13:21:08 -07:00
slice_op.h	[caffe2] SliceOp axes indexing fixes. (#45432)	2020-10-06 13:21:08 -07:00
softmax_op.cc
softmax_op.h
softmax_op_cudnn.cc
softmax_ops.cu
softmax_utils.cc
softmax_utils.h
softmax_with_loss_op.cc
softmax_with_loss_op.h
softplus_op.cc
softplus_op.cu
softplus_op.h
softsign_op.cc
softsign_op.cu
softsign_op.h
space_batch_op.cc
space_batch_op.h
space_batch_op_gpu.cu
sparse_dropout_with_replacement_op.cc	caffe2: use at::mt19937 instead of std::mt19937 (10x speedup) (#43987)	2020-10-16 16:08:35 -07:00
sparse_dropout_with_replacement_op.h
sparse_lp_regularizer_op.cc	Adding sparse Lp regularization operator to Caffe2 (#38574)	2020-06-01 15:21:19 -07:00
sparse_lp_regularizer_op.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
sparse_lp_regularizer_op_gpu.cu	Adding sparse Lp regularization operator to Caffe2 (#38574)	2020-06-01 15:21:19 -07:00
sparse_normalize_op.cc	[caffe2] Add operator schema for FP16SparseNorm (#46300)	2020-10-13 18:58:23 -07:00
sparse_normalize_op.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
sparse_normalize_op_gpu.cu
sparse_to_dense_mask_op.cc	Exposes SparseToDenseMask Caffe2 Operator (#45670)	2020-10-02 10:05:13 -07:00
sparse_to_dense_mask_op.h	Exposes SparseToDenseMask Caffe2 Operator (#45670)	2020-10-02 10:05:13 -07:00
sparse_to_dense_op.cc	Shape inference for SparseToDense in ExpertCombiner	2020-07-15 08:04:48 -07:00
sparse_to_dense_op.cu
sparse_to_dense_op.h
spatial_batch_norm_gradient_op.cc
spatial_batch_norm_op.cc
spatial_batch_norm_op.cu
spatial_batch_norm_op.h	[Caffe2] Remove explicitly divide by zero in SpatialBN training mode (#42380)	2020-08-01 11:54:58 -07:00
spatial_batch_norm_op_cudnn.cu
spatial_batch_norm_op_impl.cuh	[Caffe2] Remove explicitly divide by zero in SpatialBN training mode (#42380)	2020-08-01 11:54:58 -07:00
spatial_softmax_with_loss_op.cc
spatial_softmax_with_loss_op.h
sqr_op.cc
sqr_op.h
sqr_op_gpu.cc
sqrt_op.cc	Histogram Binning Calibration	2020-09-06 17:11:16 -07:00
sqrt_op.h
sqrt_op_gpu.cc
square_root_divide_op.cc
square_root_divide_op.h
stats_ops.cc
stats_put_ops.cc
stats_put_ops.h
stop_gradient.cc
stop_gradient.h
stop_gradient_gpu.cc
string_ops.cc	[C2] Add string equality operator (#45886)	2020-10-06 12:08:26 -07:00
string_ops.h
string_ops_test.cc
stump_func_op.cc	[dt] [caffe2] add/fix shape inference for StumpFunc, SliceGradient and ResizeLike (#35430)	2020-03-26 17:50:32 -07:00
stump_func_op.cu
stump_func_op.h
stylizer_ops.cc
summarize_op.cc
summarize_op.cu
summarize_op.h
swish_op.cc
swish_op.cu
swish_op.h
tan_op.cc
tan_op.cu
tan_op.h
tanh_gradient_op.cc
tanh_op.cc	[caffe2] fix type and shape inference for common gradient ops (#35857)	2020-04-02 11:17:04 -07:00
tanh_op.cu
tanh_op.h
tanh_op_cudnn.cc
tensor_protos_db_input.cc
tensor_protos_db_input.h
tensor_protos_db_input_gpu.cc
text_file_reader.cc
text_file_reader_utils.cc	fix more errors (#34480)	2020-03-09 14:54:15 -07:00
text_file_reader_utils.h	Renaming CAFFE2_API to TORCH_API (#49496)	2020-12-18 10:54:50 -08:00
text_file_reader_utils_test.cc
thresholded_relu_op.cc
thresholded_relu_op.cu
thresholded_relu_op.h
tile_op.cc	pass TypeMeta by value (#45026)	2020-10-30 10:14:17 -07:00
tile_op.cu
tile_op.h
top_k.cc
top_k.cu
top_k.h
top_k_heap_selection.cuh	RIP CUDA <9.2: circleci, aten, and caffe2 (#36846)	2020-05-18 13:41:05 -07:00
top_k_radix_selection.cuh	Followup for cuda assert cleanups (#39220)	2020-05-29 11:53:46 -07:00
transpose_op.cc
transpose_op.cu
transpose_op.h
transpose_op_cudnn.cc
tt_linear_op.cc
tt_linear_op.h
unique_ops.cc
unique_ops.cu
unique_ops.h
unsafe_coalesce.cc	[C2] Revive unsafe CoalesceOp (#49402)	2020-12-17 04:31:29 -08:00
unsafe_coalesce.cu	[C2] Revive unsafe CoalesceOp (#49402)	2020-12-17 04:31:29 -08:00
unsafe_coalesce.h	[C2] Revive unsafe CoalesceOp (#49402)	2020-12-17 04:31:29 -08:00
upsample_op.cc
upsample_op.cu
upsample_op.h
utility_ops.cc	Implement LengthsToOffsets operator in Caffe2 (#46590)	2020-10-29 07:03:34 -07:00
utility_ops.cu	restore at::Half support for caffe2 SumOp (#41952)	2020-07-24 10:49:06 -07:00
utility_ops.h	Implement LengthsToOffsets operator in Caffe2 (#46590)	2020-10-29 07:03:34 -07:00
utility_ops_gpu_test.cc
utility_ops_test.cc
variable_length_sequence_padding.cc
variable_length_sequence_padding.h
weighted_multi_sampling_op.cc
weighted_multi_sampling_op.h
weighted_sample_op.cc
weighted_sample_op.cu
weighted_sample_op.h
while_op.cc
while_op.h
while_op_gpu.cc
workspace_ops.cc
zero_gradient_op.cc
zero_gradient_op.h
zero_gradient_op_gpu.cc