| .. |
|
experimental/c10/cpu
|
Rename TensorTypeId to DispatchKey (#32154)
|
2020-01-15 11:16:08 -08:00 |
|
hip
|
[ROCm] Enable 3D convolutions through ROCm (#33067)
|
2020-02-14 13:19:10 -08:00 |
|
quantized
|
Fix 'template' keyword warning with clang-cl and clang.exe (#32104)
|
2020-01-14 20:09:35 -08:00 |
|
rnn
|
Fix spelling errors
|
2020-01-28 04:46:15 -08:00 |
|
abs_op.cc
|
|
|
|
abs_op.cu
|
|
|
|
abs_op.h
|
|
|
|
accumulate_op.cc
|
|
|
|
accumulate_op.cu
|
|
|
|
accumulate_op.h
|
|
|
|
accuracy_op.cc
|
|
|
|
accuracy_op.cu
|
|
|
|
accuracy_op.h
|
|
|
|
acos_op.cc
|
|
|
|
acos_op.cu
|
|
|
|
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
|
|
|
|
affine_channel_op.cu
|
|
|
|
affine_channel_op.h
|
|
|
|
alias_with_name.cc
|
move AliasWithNameOp to caffe2/operators
|
2019-12-17 02:39:40 -08: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
|
|
|
|
apmeter_op.h
|
|
|
|
arg_ops.cc
|
|
|
|
arg_ops.cu
|
|
|
|
arg_ops.h
|
|
|
|
asin_op.cc
|
|
|
|
asin_op.cu
|
|
|
|
asin_op.h
|
|
|
|
assert_op.cc
|
Fix spelling errors
|
2020-01-28 04:46:15 -08:00 |
|
assert_op.cu
|
|
|
|
assert_op.h
|
|
|
|
atan_op.cc
|
|
|
|
atan_op.cu
|
|
|
|
atan_op.h
|
|
|
|
atomic_ops.cc
|
Add 64bit atomic fetch add (#32354)
|
2020-01-17 11:43:43 -08:00 |
|
batch_box_cox_op.cc
|
|
|
|
batch_box_cox_op.h
|
|
|
|
batch_bucketize_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
batch_bucketize_op.h
|
|
|
|
batch_gather_ops.cc
|
|
|
|
batch_gather_ops.cu
|
|
|
|
batch_gather_ops.h
|
|
|
|
batch_matmul_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
batch_matmul_op.cu
|
|
|
|
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
|
move BatchPermutationOp to caffe2/operators
|
2019-12-17 14:58:27 -08:00 |
|
batch_permutation_op.cu
|
move BatchPermutationOp to caffe2/operators
|
2019-12-17 14:58:27 -08:00 |
|
batch_permutation_op.h
|
move BatchPermutationOp to caffe2/operators
|
2019-12-17 14:58:27 -08:00 |
|
batch_permutation_op_gpu_test.cc
|
move BatchPermutationOp to caffe2/operators
|
2019-12-17 14:58:27 -08:00 |
|
batch_sparse_to_dense_op.cc
|
Add cuda version for operators BatchSparseToDense and BatchDenseToSparse (#29166)
|
2019-11-05 13:06:23 -08:00 |
|
batch_sparse_to_dense_op.cu
|
Add cuda version for operators BatchSparseToDense and BatchDenseToSparse (#29166)
|
2019-11-05 13:06:23 -08:00 |
|
batch_sparse_to_dense_op.h
|
Add cuda version for operators BatchSparseToDense and BatchDenseToSparse (#29166)
|
2019-11-05 13:06:23 -08:00 |
|
bbox_transform_op.cc
|
|
|
|
bbox_transform_op.h
|
|
|
|
bisect_percentile_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
bisect_percentile_op.h
|
|
|
|
boolean_mask_ops.cc
|
|
|
|
boolean_mask_ops.cu
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
boolean_mask_ops.h
|
|
|
|
boolean_unmask_ops.cc
|
|
|
|
boolean_unmask_ops.cu
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
boolean_unmask_ops.h
|
|
|
|
boolean_unmask_ops_test.cc
|
|
|
|
box_with_nms_limit_op.cc
|
enable "keeps" from BoxWithNMSLimit and caffe2_fastrcnn_outputs_inference
|
2019-08-19 10:54:22 -07:00 |
|
box_with_nms_limit_op.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
bucketize_op.cc
|
[C2] Native GPU implementation for bucketize (#33529)
|
2020-02-21 15:47:04 -08:00 |
|
bucketize_op.cu
|
[C2] Native GPU implementation for bucketize (#33529)
|
2020-02-21 15:47:04 -08:00 |
|
bucketize_op.h
|
[C2] Native GPU implementation for bucketize (#33529)
|
2020-02-21 15:47:04 -08:00 |
|
byte_weight_dequant_op.cc
|
|
|
|
byte_weight_dequant_op.h
|
|
|
|
cast_op.cc
|
|
|
|
cast_op.cu
|
|
|
|
cast_op.h
|
|
|
|
cbrt_op.cc
|
|
|
|
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
|
batch size 0 support in ChannelShuffle DNNLOWP op (#26858)
|
2019-09-26 00:40:07 -07:00 |
|
channel_shuffle_op.cu
|
|
|
|
channel_shuffle_op.h
|
|
|
|
channel_stats_op.cc
|
|
|
|
channel_stats_op.cu
|
|
|
|
channel_stats_op.h
|
|
|
|
channelwise_conv3d_op_cudnn.cu
|
[caffe2] fix field initialization after base Clang errors (#33556)
|
2020-02-28 15:42:49 -08:00 |
|
clip_op.cc
|
|
|
|
clip_op.cu
|
|
|
|
clip_op.h
|
|
|
|
CMakeLists.txt
|
[pytorch] fix BUILD_CAFFE2_MOBILE gating around caffe2/operators/experimental/c10/cpu (#34354)
|
2020-03-09 12:00:57 -07:00 |
|
collect_and_distribute_fpn_rpn_proposals_op.cc
|
|
|
|
collect_and_distribute_fpn_rpn_proposals_op.h
|
|
|
|
communicator_op.cc
|
|
|
|
communicator_op_gpu.cc
|
|
|
|
concat_split_op.cc
|
Fix input_channels divisibility check in concat_split_op (#31448)
|
2019-12-20 09:12:54 -08:00 |
|
concat_split_op.h
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
concat_split_op_gpu.cc
|
|
|
|
conditional_op.cc
|
|
|
|
conditional_op.h
|
|
|
|
conv_gradient_op.cc
|
|
|
|
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
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08: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
|
|
|
|
conv_transpose_gradient_op.cc
|
|
|
|
conv_transpose_op.cc
|
|
|
|
conv_transpose_op.h
|
|
|
|
conv_transpose_op_cudnn.cc
|
fix zero-batch handling in convtranspose (#24341)
|
2019-12-18 15:06:36 -08:00 |
|
conv_transpose_op_gpu.cc
|
|
|
|
conv_transpose_op_impl.h
|
fix zero-batch handling in convtranspose (#24341)
|
2019-12-18 15:06:36 -08:00 |
|
conv_transpose_op_mobile.cc
|
|
|
|
conv_transpose_op_mobile.h
|
|
|
|
conv_transpose_op_mobile_impl.h
|
fix zero-batch handling in convtranspose (#24341)
|
2019-12-18 15:06:36 -08:00 |
|
conv_transpose_op_mobile_test.cc
|
Redundant condition (#32396)
|
2020-01-21 11:30:49 -08:00 |
|
conv_transpose_unpool_op_base.h
|
fix zero-batch handling in convtranspose (#24341)
|
2019-12-18 15:06:36 -08:00 |
|
copy_op.cc
|
exposing CPU/GPU Copy ops (#32248)
|
2020-01-17 12:40:43 -08: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
|
|
|
|
copy_rows_to_tensor_op.h
|
|
|
|
cos_op.cc
|
|
|
|
cos_op.cu
|
|
|
|
cos_op.h
|
|
|
|
cosh_op.cc
|
|
|
|
cosh_op.cu
|
|
|
|
cosh_op.h
|
|
|
|
cosine_embedding_criterion_op.cc
|
|
|
|
cosine_embedding_criterion_op.cu
|
|
|
|
cosine_embedding_criterion_op.h
|
|
|
|
counter_ops.cc
|
Replace c10::guts::stuff with std::stuff (#30915)
|
2019-12-16 13:57:19 -08:00 |
|
counter_ops.h
|
|
|
|
counter_ops_gpu.cc
|
|
|
|
crash_op.cc
|
|
|
|
create_scope_op.cc
|
|
|
|
create_scope_op.h
|
|
|
|
crf_viterbi_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
cross_entropy_op.cc
|
|
|
|
cross_entropy_op.cu
|
|
|
|
cross_entropy_op.h
|
|
|
|
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
|
|
|
|
dataset_ops.h
|
|
|
|
deform_conv_gradient_op.cc
|
|
|
|
deform_conv_op.cc
|
|
|
|
deform_conv_op.cu
|
|
|
|
deform_conv_op.h
|
|
|
|
deform_conv_op_impl.h
|
Back out "Revert D17908478: Switch PyTorch/Caffe2 to C++14"
|
2019-12-03 14:33:43 -08:00 |
|
dense_vector_to_id_list_op.cc
|
|
|
|
dense_vector_to_id_list_op.h
|
|
|
|
depthwise_3x3_conv_op_cudnn.cu
|
[caffe2] fix field initialization after base Clang errors (#33556)
|
2020-02-28 15:42:49 -08:00 |
|
distance_op.cc
|
|
|
|
distance_op.cu
|
|
|
|
distance_op.h
|
|
|
|
do_op.cc
|
|
|
|
do_op.h
|
|
|
|
do_op_gpu.cc
|
|
|
|
dropout_op.cc
|
|
|
|
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
|
|
|
|
elementwise_div_op.cc
|
|
|
|
elementwise_div_op.cu
|
|
|
|
elementwise_div_op.h
|
|
|
|
elementwise_linear_op.cc
|
|
|
|
elementwise_linear_op.cu
|
|
|
|
elementwise_linear_op.h
|
|
|
|
elementwise_logical_ops.cc
|
|
|
|
elementwise_logical_ops.h
|
|
|
|
elementwise_mul_gradient_op.cc
|
optimize MulGradient for common shapes (#19705)
|
2019-12-11 11:39:52 -08:00 |
|
elementwise_mul_op.cc
|
|
|
|
elementwise_mul_op.cu
|
[C2] Small improvement for elementwise_mul operator. (#33537)
|
2020-02-24 23:00:27 -08:00 |
|
elementwise_mul_op.h
|
optimize MulGradient for common shapes (#19705)
|
2019-12-11 11:39:52 -08:00 |
|
elementwise_op_gpu_test.cc
|
|
|
|
elementwise_op_test.cc
|
|
|
|
elementwise_op_test.h
|
|
|
|
elementwise_ops.cc
|
|
|
|
elementwise_ops.cu
|
|
|
|
elementwise_ops.h
|
|
|
|
elementwise_ops_schema.cc
|
[Caffe2] Fix shape inference for element-wise operators (#33431)
|
2020-02-25 09:03:06 -08:00 |
|
elementwise_ops_utils.cc
|
|
|
|
elementwise_ops_utils.h
|
|
|
|
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
|
Update the descriptive error message for enforce fail (#31575)
|
2020-01-03 13:53:20 -08:00 |
|
ensure_clipped_op.cc
|
|
|
|
ensure_clipped_op.h
|
|
|
|
ensure_cpu_output_op.cc
|
|
|
|
ensure_cpu_output_op.cu
|
|
|
|
ensure_cpu_output_op.h
|
|
|
|
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
|
|
|
|
expand_op_gpu.cc
|
|
|
|
expand_squeeze_dims_op.cc
|
|
|
|
expand_squeeze_dims_op.h
|
|
|
|
expand_squeeze_dims_op_gpu.cc
|
|
|
|
fc_inference.cc
|
|
|
|
fc_inference.h
|
|
|
|
feature_maps_ops.cc
|
|
|
|
feature_maps_ops.h
|
|
|
|
feed_blob_op.cc
|
|
|
|
feed_blob_op.h
|
|
|
|
filler_op.cc
|
|
|
|
filler_op.cu
|
|
|
|
filler_op.h
|
Replace c10::guts::stuff with std::stuff (#30915)
|
2019-12-16 13:57:19 -08:00 |
|
find_duplicate_elements_op.cc
|
|
|
|
find_duplicate_elements_op.h
|
|
|
|
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
|
|
|
|
fully_connected_op.h
|
[caffe2] remove dnnlowp log code (#33184)
|
2020-02-12 00:47:09 -08:00 |
|
fully_connected_op_gpu.cc
|
|
|
|
fused_rowwise_8bit_conversion_ops.cc
|
[caffe2] make fused rowwise quant/dequant op work for N-dim tensors (#33426)
|
2020-02-19 23:29:42 -08:00 |
|
fused_rowwise_8bit_conversion_ops.h
|
[caffe2] make fused rowwise quant/dequant op work for N-dim tensors (#33426)
|
2020-02-19 23:29:42 -08:00 |
|
fused_rowwise_random_quantization_ops.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
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
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
gather_ranges_to_dense_op.cc
|
[caffe2] Add embedding empty ratio checker (disabled by default) (#33145)
|
2020-02-24 16:10:01 -08:00 |
|
gather_ranges_to_dense_op.h
|
[caffe2] Add embedding empty ratio checker (disabled by default) (#33145)
|
2020-02-24 16:10:01 -08:00 |
|
gelu_op.cc
|
preprocessor cleanup (#33957)
|
2020-03-02 13:37:19 -08:00 |
|
gelu_op.cu
|
|
|
|
gelu_op.h
|
|
|
|
generate_proposals_op.cc
|
|
|
|
generate_proposals_op.cu
|
switch to rocThrust for thrust/cub APIs (#25620)
|
2019-09-03 22:16:30 -07:00 |
|
generate_proposals_op.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -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
|
Fix rotated rect intersection error (#24171)
|
2019-08-13 16:23:13 -07:00 |
|
generate_proposals_op_util_nms_gpu.cu
|
|
|
|
generate_proposals_op_util_nms_gpu.h
|
|
|
|
generate_proposals_op_util_nms_gpu_test.cc
|
Skips flaky UtilsNMSTest.GPUEqualsCPURotatedCorrectnessTest (#30053)
|
2019-11-21 13:44:44 -08:00 |
|
generate_proposals_op_util_nms_test.cc
|
Fix rotated rect intersection error (#24171)
|
2019-08-13 16:23:13 -07:00 |
|
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
|
|
|
|
given_tensor_fill_op.cc
|
|
|
|
given_tensor_fill_op.cu
|
|
|
|
given_tensor_fill_op.h
|
|
|
|
glu_op.cc
|
|
|
|
glu_op.cu
|
|
|
|
glu_op.h
|
|
|
|
group_norm_op.cc
|
|
|
|
group_norm_op.cu
|
|
|
|
group_norm_op.h
|
batch size 0 support in norm operators (#26894)
|
2019-09-26 16:08:35 -07:00 |
|
gru_unit_op.cc
|
|
|
|
gru_unit_op.h
|
|
|
|
gru_unit_op_gpu.cu
|
|
|
|
h_softmax_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
h_softmax_op.h
|
|
|
|
half_float_ops.cc
|
Back out "Use simd version for fp16 conversions" (#32640)
|
2020-01-27 10:01:24 -08:00 |
|
half_float_ops.cu
|
|
|
|
half_float_ops.h
|
call fp16<->fp32 routines in fbgemm from Half2Float and Float2Half operators (#30715)
|
2019-12-07 19:46:47 -08:00 |
|
half_float_ops_test.cc
|
|
|
|
hard_sigmoid_op.cc
|
|
|
|
hard_sigmoid_op.cu
|
|
|
|
hard_sigmoid_op.h
|
|
|
|
heatmap_max_keypoint_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
heatmap_max_keypoint_op.h
|
register HeatmapMaxKeypoint with C10 (#25191)
|
2019-08-27 20:13:57 -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
|
[1/3] Bind IndexHash to PyTorch (#33015)
|
2020-02-10 17:47:38 -08:00 |
|
index_hash_ops.h
|
[1/3] Bind IndexHash to PyTorch (#33015)
|
2020-02-10 17:47:38 -08:00 |
|
index_ops.cc
|
|
|
|
index_ops.h
|
|
|
|
inference_lstm_op.cc
|
|
|
|
inference_lstm_op.h
|
|
|
|
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
|
|
|
|
layer_norm_op.cc
|
|
|
|
layer_norm_op.cu
|
|
|
|
layer_norm_op.h
|
|
|
|
leaky_relu_op.cc
|
|
|
|
leaky_relu_op.cu
|
|
|
|
leaky_relu_op.h
|
|
|
|
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
|
|
|
|
lengths_reducer_fused_8bit_rowwise_ops.h
|
[caffe2] use 2-stage EmbeddingSpMDM interface (#32271)
|
2020-01-22 19:05:36 -08:00 |
|
lengths_reducer_ops.cc
|
|
|
|
lengths_reducer_ops.h
|
[caffe2] use JIT'ed fp16 SLS (#32432)
|
2020-02-13 21:15:39 -08:00 |
|
lengths_reducer_rowwise_8bit_ops.cc
|
|
|
|
lengths_reducer_rowwise_8bit_ops.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08: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
|
SeqBlobReader Implementation (#29888)
|
2019-11-16 01:18:54 -08:00 |
|
load_save_op_gpu.cc
|
|
|
|
load_save_op_util.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
load_save_op_util.h
|
Add missing _API definitions. (#30310)
|
2019-12-03 10:46:32 -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
|
|
|
|
log_op.cc
|
|
|
|
log_op.h
|
|
|
|
log_op_gpu.cc
|
|
|
|
logit_op.cc
|
|
|
|
logit_op.cu
|
|
|
|
logit_op.h
|
|
|
|
loss_op.cc
|
|
|
|
loss_op.cu
|
|
|
|
loss_op.h
|
|
|
|
lp_pool_op.cc
|
|
|
|
lp_pool_op.cu
|
|
|
|
lpnorm_op.cc
|
|
|
|
lpnorm_op.h
|
|
|
|
lstm_unit_op.cc
|
|
|
|
lstm_unit_op.h
|
|
|
|
lstm_unit_op_gpu.cu
|
|
|
|
lstm_utils.h
|
|
|
|
map_ops.cc
|
|
|
|
map_ops.h
|
|
|
|
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
|
|
|
|
mem_query_op.cu
|
|
|
|
merge_id_lists_op.cc
|
|
|
|
merge_id_lists_op.h
|
|
|
|
minmax_gradient_ops.cc
|
|
|
|
minmax_ops.cc
|
|
|
|
minmax_ops.cu
|
|
|
|
minmax_ops.h
|
|
|
|
mod_op.cc
|
|
|
|
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
|
|
|
|
normalize_op.cc
|
|
|
|
normalize_op.h
|
|
|
|
normalize_ops.cu
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
numpy_tile_op.cc
|
|
|
|
numpy_tile_op.h
|
|
|
|
one_hot_ops.cc
|
|
|
|
one_hot_ops.cu
|
|
|
|
one_hot_ops.h
|
|
|
|
onnx_while_op.cc
|
|
|
|
onnx_while_op.h
|
|
|
|
op_utils_cudnn.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
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
|
Change PackSegments to ensure consistent behavior between CPU and GPU
|
2019-12-30 13:31:45 -08:00 |
|
pack_segments.cu
|
|
|
|
pack_segments.h
|
|
|
|
pad_op.cc
|
|
|
|
pad_op.h
|
|
|
|
pad_op_gpu.cu
|
|
|
|
partition_ops.cc
|
Implement gradient operator for GatherByKeys. (#24348)
|
2019-08-15 12:19:22 -07:00 |
|
partition_ops.h
|
|
|
|
percentile_op.cc
|
|
|
|
percentile_op.h
|
|
|
|
perplexity_op.cc
|
|
|
|
perplexity_op.cu
|
|
|
|
perplexity_op.h
|
|
|
|
piecewise_linear_transform_op.cc
|
Expose PiecewiseLinearTransform to PyTorch
|
2019-09-27 12:49:04 -07:00 |
|
piecewise_linear_transform_op.cu
|
Expose PiecewiseLinearTransform to PyTorch
|
2019-09-27 12:49:04 -07:00 |
|
piecewise_linear_transform_op.h
|
Expose PiecewiseLinearTransform to PyTorch
|
2019-09-27 12:49:04 -07:00 |
|
pool_gradient_op.cc
|
|
|
|
pool_op.cc
|
|
|
|
pool_op.cu
|
|
|
|
pool_op.h
|
|
|
|
pool_op_cudnn.cc
|
|
|
|
pool_op_util.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
pool_op_util.h
|
|
|
|
pow_op.cc
|
Pow() : Use lightweight operations for predefined scalar exponent values (#28903)
|
2019-10-31 05:39:39 -07:00 |
|
pow_op.cu
|
|
|
|
pow_op.h
|
|
|
|
prefetch_op.h
|
|
|
|
prelu_op.cc
|
|
|
|
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
|
|
|
|
quantile_op.cc
|
[Rowwise Pruning][c2 op] Add Quantile Op (#32448)
|
2020-01-22 16:59:56 -08:00 |
|
quantile_op.h
|
[Rowwise Pruning][c2 op] Add Quantile Op (#32448)
|
2020-01-22 16:59:56 -08:00 |
|
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
|
switch to rocThrust for thrust/cub APIs (#25620)
|
2019-09-03 22:16:30 -07:00 |
|
reduce_front_back_max_ops.h
|
|
|
|
reduce_front_back_mean_ops.cc
|
|
|
|
reduce_front_back_sum_mean_ops.cu
|
|
|
|
reduce_front_back_sum_mean_ops.h
|
|
|
|
reduce_front_back_sum_ops.cc
|
|
|
|
reduce_ops.cc
|
Add a loop test for onnxified net (#32935)
|
2020-02-03 18:35:41 -08:00 |
|
reduce_ops.cu
|
|
|
|
reduce_ops.h
|
|
|
|
reducer_functors.h
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
reduction_ops.cc
|
|
|
|
reduction_ops.cu
|
|
|
|
reduction_ops.h
|
Remove Tensor.h, TensorMethods.h from src/core. (#27086)
|
2019-10-06 09:37:50 -07:00 |
|
relu_n_op.cc
|
|
|
|
relu_n_op.cu
|
|
|
|
relu_n_op.h
|
|
|
|
relu_op.cc
|
|
|
|
relu_op.cu
|
|
|
|
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
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
reshape_op.cc
|
|
|
|
reshape_op.h
|
[C2] Fix slowness of the ReshapeOp. (#33729)
|
2020-03-03 00:44:22 -08:00 |
|
reshape_op_gpu.cc
|
|
|
|
reshape_op_gpu_test.cc
|
|
|
|
resize_3d_op.cc
|
Migrate the cpu and gpu implementations of resize nearest 3D from vision to caffe2
|
2019-10-03 16:14:00 -07:00 |
|
resize_3d_op.cu
|
Migrate the cpu and gpu implementations of resize nearest 3D from vision to caffe2
|
2019-10-03 16:14:00 -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
|
|
|
|
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
|
switch to rocThrust for thrust/cub APIs (#25620)
|
2019-09-03 22:16:30 -07:00 |
|
rmac_regions_op.h
|
|
|
|
roi_align_gradient_op.cc
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
roi_align_gradient_op.cu
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
roi_align_gradient_op.h
|
|
|
|
roi_align_op.cc
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
roi_align_op.cu
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
roi_align_op.h
|
|
|
|
roi_align_op_gpu_test.cc
|
|
|
|
roi_align_rotated_gradient_op.cc
|
|
|
|
roi_align_rotated_gradient_op.cu
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
roi_align_rotated_gradient_op.h
|
|
|
|
roi_align_rotated_op.cc
|
preprocessor cleanup (#33957)
|
2020-03-02 13:37:19 -08:00 |
|
roi_align_rotated_op.cu
|
Register RoIAlignRotated with C10
|
2020-01-16 16:32:28 -08:00 |
|
roi_align_rotated_op.h
|
Register RoIAlignRotated with C10
|
2020-01-16 16:32:28 -08:00 |
|
roi_pool_op.cc
|
|
|
|
roi_pool_op.cu
|
|
|
|
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
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
scale_blobs_op.h
|
|
|
|
scale_op.cc
|
|
|
|
scale_op.h
|
|
|
|
scale_op_gpu.cc
|
|
|
|
segment_reduction_op.cc
|
|
|
|
segment_reduction_op.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
segment_reduction_op_gpu.cu
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
selu_op.cc
|
|
|
|
selu_op.cu
|
|
|
|
selu_op.h
|
|
|
|
sequence_ops.cc
|
|
|
|
sequence_ops.cu
|
[caffe2] fix no matching function min/max Clang errors (#33563)
|
2020-02-28 11:33:24 -08:00 |
|
sequence_ops.h
|
|
|
|
shape_op.cc
|
|
|
|
shape_op.h
|
|
|
|
shape_op_gpu.cc
|
|
|
|
sigmoid_gradient_op.cc
|
|
|
|
sigmoid_op.cc
|
|
|
|
sigmoid_op.cu
|
|
|
|
sigmoid_op.h
|
|
|
|
sigmoid_op_cudnn.cc
|
|
|
|
sin_op.cc
|
|
|
|
sin_op.cu
|
|
|
|
sin_op.h
|
|
|
|
sinh_op.cc
|
|
|
|
sinh_op.cu
|
|
|
|
sinh_op.h
|
|
|
|
sinusoid_position_encoding_op.cc
|
|
|
|
sinusoid_position_encoding_op.h
|
preprocessor cleanup (#33957)
|
2020-03-02 13:37:19 -08:00 |
|
slice_op.cc
|
|
|
|
slice_op.cu
|
|
|
|
slice_op.h
|
|
|
|
softmax_op.cc
|
Support softmax with D == 0 (#29167)
|
2019-11-11 00:46:10 -08:00 |
|
softmax_op.h
|
|
|
|
softmax_op_cudnn.cc
|
Support softmax with D == 0 (#29167)
|
2019-11-11 00:46:10 -08:00 |
|
softmax_ops.cu
|
Support softmax with D == 0 (#29167)
|
2019-11-11 00:46:10 -08:00 |
|
softmax_utils.cc
|
|
|
|
softmax_utils.h
|
|
|
|
softmax_with_loss_op.cc
|
Linearizable Label: Class Weights, Allow Missing Label, and Average by Batch Size (#29707)
|
2019-11-13 16:52:27 -08: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
|
|
|
|
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
|
|
|
|
sparse_dropout_with_replacement_op.h
|
|
|
|
sparse_normalize_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
sparse_normalize_op.h
|
Remove gradient value as input from SparseNormalize op (#24357)
|
2019-08-19 14:47:09 -07:00 |
|
sparse_normalize_op_gpu.cu
|
|
|
|
sparse_to_dense_mask_op.cc
|
|
|
|
sparse_to_dense_mask_op.h
|
|
|
|
sparse_to_dense_op.cc
|
|
|
|
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
|
Fix for batch norm 2D with affine=False (#29458)
|
2019-11-18 21:52:11 -08:00 |
|
spatial_batch_norm_op_cudnn.cu
|
|
|
|
spatial_batch_norm_op_impl.cuh
|
|
|
|
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
|
|
|
|
sqrt_op.h
|
|
|
|
sqrt_op_gpu.cc
|
|
|
|
square_root_divide_op.cc
|
|
|
|
square_root_divide_op.h
|
[caffe2] Fix signed unsigned comparison warning (#34161)
|
2020-03-04 08:02:44 -08:00 |
|
stats_ops.cc
|
|
|
|
stats_put_ops.cc
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
stats_put_ops.h
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
stop_gradient.cc
|
|
|
|
stop_gradient.h
|
|
|
|
stop_gradient_gpu.cc
|
|
|
|
string_ops.cc
|
|
|
|
string_ops.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
string_ops_test.cc
|
Replace c10::guts::stuff with std::stuff (#30915)
|
2019-12-16 13:57:19 -08:00 |
|
stump_func_op.cc
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
stump_func_op.cu
|
|
|
|
stump_func_op.h
|
|
|
|
stylizer_ops.cc
|
|
|
|
summarize_op.cc
|
|
|
|
summarize_op.cu
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
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
|
|
|
|
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
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08:00 |
|
text_file_reader_utils.cc
|
fix more errors (#34480)
|
2020-03-09 14:54:15 -07:00 |
|
text_file_reader_utils.h
|
|
|
|
text_file_reader_utils_test.cc
|
|
|
|
thresholded_relu_op.cc
|
|
|
|
thresholded_relu_op.cu
|
|
|
|
thresholded_relu_op.h
|
|
|
|
tile_op.cc
|
|
|
|
tile_op.cu
|
|
|
|
tile_op.h
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
top_k.cc
|
|
|
|
top_k.cu
|
|
|
|
top_k.h
|
|
|
|
top_k_heap_selection.cuh
|
|
|
|
top_k_radix_selection.cuh
|
|
|
|
transpose_op.cc
|
|
|
|
transpose_op.cu
|
|
|
|
transpose_op.h
|
Add Int8Transpose operator (#16382)
|
2019-08-29 16:06:25 -07:00 |
|
transpose_op_cudnn.cc
|
Disable cudnn transpose for int types (#26934)
|
2019-09-27 11:36:10 -07:00 |
|
tt_linear_op.cc
|
|
|
|
tt_linear_op.h
|
|
|
|
unique_ops.cc
|
|
|
|
unique_ops.cu
|
switch to rocThrust for thrust/cub APIs (#25620)
|
2019-09-03 22:16:30 -07:00 |
|
unique_ops.h
|
|
|
|
upsample_op.cc
|
|
|
|
upsample_op.cu
|
|
|
|
upsample_op.h
|
|
|
|
utility_ops.cc
|
Extend HasElements to support multiple inputs (#28717)
|
2019-10-27 23:00:07 -07:00 |
|
utility_ops.cu
|
Fix typos, via a Levenshtein-type corrector (#31523)
|
2020-01-17 16:03:19 -08:00 |
|
utility_ops.h
|
Fix typos (#30606)
|
2019-12-02 20:17:42 -08: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
|
|
|
