OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2026-01-15 12:15:51 +00:00
Code Issues Actions 52 Packages Projects Releases Wiki Activity
Files
7c825bad10fa81e330f8ecd4e8cac5aa17143b21
pytorch/caffe2/operators
History
Yuxin Wu 9a4bc67f66 [caffe2/detectron2] fix Mask R-CNN caffe2 conversion on GPU (#36366) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/36366

fix issues introduced in D20528758

Reviewed By: linbinyu

Differential Revision: D20959367

fbshipit-source-id: 920055a6782b9c6729177f7101f2f9eb3e40ebf8
2020-04-10 01:12:13 -07:00
..
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
Formatting cmake (to lowercase without space for if/elseif/else/endif) (#35521)
2020-03-27 14:25:17 -07: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
…
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
…
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
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_gpu_test.cc
…
batch_matmul_op_test.cc
…
batch_matmul_op.cc
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
batch_matmul_op.cu
…
batch_matmul_op.h
…
batch_moments_op.cc
…
batch_moments_op.cu
…
batch_moments_op.h
…
batch_permutation_op_gpu_test.cc
…
batch_permutation_op.cc
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
batch_permutation_op.cu
…
batch_permutation_op.h
…
batch_sparse_to_dense_op.cc
…
batch_sparse_to_dense_op.cu
…
batch_sparse_to_dense_op.h
…
bbox_transform_op.cc
[3] register caffe2 mask rcnn ops in lite interpreter (#35246)
2020-03-26 16:42:29 -07:00
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
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
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_test.cc
…
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
…
box_with_nms_limit_op.cc
[3] register caffe2 mask rcnn ops in lite interpreter (#35246)
2020-03-26 16:42:29 -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
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
cast_op.cc
…
cast_op.cu
…
cast_op.h
…
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
…
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
Fix PyTorch separate compilation (Reland) (#35581)
2020-03-30 13:21:57 -07:00
collect_and_distribute_fpn_rpn_proposals_op.cc
…
collect_and_distribute_fpn_rpn_proposals_op.h
…
communicator_op_gpu.cc
…
communicator_op.cc
…
concat_split_op_gpu.cc
…
concat_split_op.cc
…
concat_split_op.h
…
conditional_op.cc
…
conditional_op.h
…
conv_gradient_op.cc
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
conv_op_cache_cudnn_test.cc
…
conv_op_cache_cudnn.cc
…
conv_op_cache_cudnn.h
…
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_gpu.cc
…
conv_op_shared.cc
…
conv_op_shared.h
…
conv_op.cc
…
conv_op.h
…
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_cudnn.cc
…
conv_transpose_op_gpu.cc
…
conv_transpose_op_impl.h
…
conv_transpose_op_mobile_impl.h
…
conv_transpose_op_mobile_test.cc
Redundant condition (#32396)
2020-01-21 11:30:49 -08:00
conv_transpose_op_mobile.cc
…
conv_transpose_op_mobile.h
…
conv_transpose_op.cc
…
conv_transpose_op.h
…
conv_transpose_unpool_op_base.h
…
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
[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_gpu.cc
…
counter_ops.cc
…
counter_ops.h
…
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_gpu.cu
…
data_couple.cc
…
data_couple.h
…
dataset_ops.cc
[caffe2] Fix typo in dataset_ops (#35356)
2020-03-24 22:24:46 -07:00
dataset_ops.h
…
deform_conv_gradient_op.cc
…
deform_conv_op_impl.h
…
deform_conv_op.cc
…
deform_conv_op.cu
…
deform_conv_op.h
…
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_gpu.cc
…
do_op.cc
…
do_op.h
…
dropout_op_cudnn.cc
…
dropout_op.cc
…
dropout_op.cu
…
dropout_op.h
…
elementwise_add_gradient_op.cc
…
elementwise_add_op_gpu.cc
…
elementwise_add_op.cc
…
elementwise_add_op.h
…
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
…
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] Small improvement for elementwise_mul operator. (#33537)
2020-02-24 23:00:27 -08:00
elementwise_mul_op.h
…
elementwise_op_gpu_test.cc
…
elementwise_op_test.cc
…
elementwise_op_test.h
…
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
…
elementwise_ops_utils.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
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
elementwise_sub_gradient_op.cc
…
elementwise_sub_op_gpu.cc
…
elementwise_sub_op.cc
…
elementwise_sub_op.h
…
elementwise_sum_op.cc
…
elu_op_cudnn.cc
…
elu_op.cc
…
elu_op.cu
…
elu_op.h
…
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
…
erf_op.cc
…
erf_op.cu
…
erf_op.h
…
exp_op_gpu.cc
…
exp_op.cc
…
exp_op.h
…
expand_op_gpu.cc
…
expand_op.cc
…
expand_op.h
…
expand_squeeze_dims_op_gpu.cc
…
expand_squeeze_dims_op.cc
…
expand_squeeze_dims_op.h
…
fc_inference.cc
…
fc_inference.h
…
feature_maps_ops.cc
…
feature_maps_ops.h
[caffe2][debuggability] add length checks to MergeMultiScalarFeatureTensors (#36248)
2020-04-09 02:57:21 -07:00
feed_blob_op.cc
…
feed_blob_op.h
…
filler_op.cc
…
filler_op.cu
…
filler_op.h
…
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_gpu.cc
…
free_op.cc
…
free_op.h
…
fully_connected_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
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_nbit_conversion_ops.cc
fp16 include not needed (#35708)
2020-03-30 17:47:44 -07:00
fused_rowwise_nbit_conversion_ops.h
[caffe2] open source 2/4-bit SLS operators (#34903)
2020-03-17 22:55:10 -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] open source 2/4-bit SLS operators (#34903)
2020-03-17 22:55:10 -07: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
[fix][tiny][caffe2] Avoid triggering errors when allow ratio is 100% (#34757)
2020-03-16 11:39:32 -07: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
…
generate_proposals_op_gpu_test.cc
…
generate_proposals_op_test.cc
…
generate_proposals_op_util_boxes_test.cc
…
generate_proposals_op_util_boxes.h
…
generate_proposals_op_util_nms_gpu_test.cc
…
generate_proposals_op_util_nms_gpu.cu
…
generate_proposals_op_util_nms_gpu.h
…
generate_proposals_op_util_nms_test.cc
…
generate_proposals_op_util_nms.h
…
generate_proposals_op.cc
[caffe2/detectron2] fix Mask R-CNN caffe2 conversion on GPU (#36366)
2020-04-10 01:12:13 -07:00
generate_proposals_op.cu
…
generate_proposals_op.h
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08: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
…
gru_unit_op_gpu.cu
…
gru_unit_op.cc
…
gru_unit_op.h
…
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_test.cc
…
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
…
hard_sigmoid_op.cc
…
hard_sigmoid_op.cu
…
hard_sigmoid_op.h
…
heatmap_max_keypoint_op.cc
[3] register caffe2 mask rcnn ops in lite interpreter (#35246)
2020-03-26 16:42:29 -07:00
heatmap_max_keypoint_op.h
…
if_op_gpu.cc
…
if_op.cc
…
if_op.h
…
im2col_op_gpu.cc
…
im2col_op.cc
…
im2col_op.h
…
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
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07:00
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
[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
…
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] 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
Split SparseLengthsSumSparse into SparseLengthsSumSparseLookup + SparseLengthsSum (#35507)
2020-03-30 13:33:29 -07:00
lengths_reducer_fused_nbit_rowwise_ops.h
[caffe2] tune prefetch distance
2020-04-04 11:40:05 -07:00
lengths_reducer_ops.cc
…
lengths_reducer_ops.h
[caffe2] explicitly pass use_offsets=false when calling fbgemm embedding kernels (#35711)
2020-03-31 08:35:19 -07: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_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
…
load_save_op.cc
…
load_save_op.h
…
local_response_normalization_op_cudnn.cc
…
local_response_normalization_op.cc
…
local_response_normalization_op.cu
…
local_response_normalization_op.h
…
locally_connected_op_gpu.cc
…
locally_connected_op_impl.h
…
locally_connected_op_util.cc
…
locally_connected_op_util.h
…
locally_connected_op.cc
…
locally_connected_op.h
…
log_op_gpu.cc
…
log_op.cc
…
log_op.h
…
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_gpu.cu
…
lstm_unit_op.cc
…
lstm_unit_op.h
[caffe2] Explicit vectorization of LSTM operator (#35556)
2020-04-01 17:19:56 -07:00
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_gpu.cc
…
matmul_op.cc
…
matmul_op.h
…
max_pool_with_index_gpu.h
…
max_pool_with_index.cu
…
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_gpu.cc
…
negate_gradient_op.cc
…
negate_gradient_op.h
…
negative_op_gpu.cc
…
negative_op.cc
…
negative_op.h
…
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_test.cc
…
operator_fallback_gpu.h
…
order_switch_ops_cudnn.cc
…
order_switch_ops_gpu.cc
…
order_switch_ops.cc
…
order_switch_ops.h
…
pack_rnn_sequence_op.cc
…
pack_rnn_sequence_op.h
…
pack_segments.cc
…
pack_segments.cu
…
pack_segments.h
…
pad_op_gpu.cu
…
pad_op.cc
…
pad_op.h
…
partition_ops.cc
…
partition_ops.h
…
percentile_op.cc
…
percentile_op.h
…
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_cudnn.cc
…
pool_op_util.cc
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
pool_op_util.h
…
pool_op.cc
…
pool_op.cu
…
pool_op.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
…
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_gpu.cc
…
prepend_dim_op.cc
…
prepend_dim_op.h
…
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
…
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
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
…
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
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
reshape_op_gpu_test.cc
…
reshape_op_gpu.cc
…
reshape_op.cc
…
reshape_op.h
[C2] Fix slowness of the ReshapeOp. (#33729)
2020-03-03 00:44:22 -08:00
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
…
roi_align_gradient_op.cc
Export roi_align_gradient_op to c10 (#34776)
2020-03-15 02:43:39 -07:00
roi_align_gradient_op.cu
Export roi_align_gradient_op to c10 (#34776)
2020-03-15 02:43:39 -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_gpu_test.cc
Disable unit test failing on Windows (#35549)
2020-03-27 19:20:29 -07:00
roi_align_op.cc
Refactor RoIAlignOp on CPU (#34698)
2020-03-27 07:53:58 -07:00
roi_align_op.cu
Refactor RoIAlignOp on CPU (#34698)
2020-03-27 07:53:58 -07:00
roi_align_op.h
Refactor RoIAlignOp on CPU (#34698)
2020-03-27 07:53:58 -07:00
roi_align_rotated_gradient_op.cc
…
roi_align_rotated_gradient_op.cu
…
roi_align_rotated_gradient_op.h
…
roi_align_rotated_op.cc
[7] add missing roi_align_rotated op to lite interpreter (#35244)
2020-03-27 07:26:02 -07: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_gpu.cc
…
scale_op.cc
…
scale_op.h
…
segment_reduction_op_gpu.cu
[caffe2] fix no matching function min/max Clang errors (#33563)
2020-02-28 11:33:24 -08:00
segment_reduction_op.cc
…
segment_reduction_op.h
Fix signed-unsigned warnings (#34791)
2020-03-19 00:29:56 -07: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_gpu.cc
…
shape_op.cc
…
shape_op.h
…
sigmoid_gradient_op.cc
…
sigmoid_op_cudnn.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
…
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
preprocessor cleanup (#33957)
2020-03-02 13:37:19 -08:00
slice_op.cc
[dt] [caffe2] add/fix shape inference for StumpFunc, SliceGradient and ResizeLike (#35430)
2020-03-26 17:50:32 -07:00
slice_op.cu
…
slice_op.h
…
softmax_op_cudnn.cc
…
softmax_op.cc
…
softmax_op.h
…
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_gpu.cu
…
space_batch_op.cc
…
space_batch_op.h
…
sparse_dropout_with_replacement_op.cc
…
sparse_dropout_with_replacement_op.h
…
sparse_normalize_op_gpu.cu
…
sparse_normalize_op.cc
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
sparse_normalize_op.h
…
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_cudnn.cu
…
spatial_batch_norm_op_impl.cuh
…
spatial_batch_norm_op.cc
…
spatial_batch_norm_op.cu
…
spatial_batch_norm_op.h
…
spatial_softmax_with_loss_op.cc
…
spatial_softmax_with_loss_op.h
…
sqr_op_gpu.cc
…
sqr_op.cc
…
sqr_op.h
…
sqrt_op_gpu.cc
…
sqrt_op.cc
…
sqrt_op.h
…
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
…
stats_put_ops.h
…
stop_gradient_gpu.cc
…
stop_gradient.cc
…
stop_gradient.h
…
string_ops_test.cc
…
string_ops.cc
…
string_ops.h
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
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
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_cudnn.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
…
tensor_protos_db_input_gpu.cc
…
tensor_protos_db_input.cc
…
tensor_protos_db_input.h
…
text_file_reader_utils_test.cc
…
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.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_heap_selection.cuh
…
top_k_radix_selection.cuh
…
top_k.cc
…
top_k.cu
…
top_k.h
…
transpose_op_cudnn.cc
…
transpose_op.cc
…
transpose_op.cu
…
transpose_op.h
…
tt_linear_op.cc
…
tt_linear_op.h
…
unique_ops.cc
…
unique_ops.cu
…
unique_ops.h
…
upsample_op.cc
…
upsample_op.cu
…
upsample_op.h
…
utility_ops_gpu_test.cc
…
utility_ops_test.cc
…
utility_ops.cc
idtt + zch distributed inference (#35763)
2020-04-03 12:09:34 -07:00
utility_ops.cu
Fix typos, via a Levenshtein-type corrector (#31523)
2020-01-17 16:03:19 -08:00
utility_ops.h
…
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_gpu.cc
…
while_op.cc
…
while_op.h
…
workspace_ops.cc
…
zero_gradient_op_gpu.cc
…
zero_gradient_op.cc
…
zero_gradient_op.h
…