pytorch/caffe2/operators
Jeff Moe e5ae8f15ad Forklet of Pytorch 2023-11-08 09:01:59 -07:00
..
hip Forklet of Pytorch 2023-11-08 09:01:59 -07:00
quantized Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rnn Forklet of Pytorch 2023-11-08 09:01:59 -07:00
CMakeLists.txt Forklet of Pytorch 2023-11-08 09:01:59 -07:00
abs_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
abs_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
abs_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accumulate_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accumulate_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accumulate_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accuracy_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accuracy_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
accuracy_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
acos_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
acos_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
acos_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
activation_ops_cudnn.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
affine_channel_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
affine_channel_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
affine_channel_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
alias_with_name.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
alias_with_name.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
alias_with_name.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
apmeter_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
apmeter_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
arg_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
arg_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
arg_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
asin_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
asin_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
asin_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
assert_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
assert_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
assert_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
async_net_barrier_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
async_net_barrier_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
async_net_barrier_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
atan_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
atan_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
atan_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
atomic_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_box_cox_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_box_cox_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_bucketize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_bucketize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_gather_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_gather_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_gather_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_matmul_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_matmul_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_matmul_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_matmul_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_matmul_op_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_moments_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_moments_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_moments_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_permutation_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_permutation_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_permutation_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_permutation_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_sparse_to_dense_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_sparse_to_dense_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
batch_sparse_to_dense_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bbox_transform_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bbox_transform_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bisect_percentile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bisect_percentile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_mask_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_mask_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_mask_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_unmask_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_unmask_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_unmask_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
boolean_unmask_ops_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
box_with_nms_limit_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
box_with_nms_limit_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bucketize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bucketize_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
bucketize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
byte_weight_dequant_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
byte_weight_dequant_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cast_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cast_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cast_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cbrt_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cbrt_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cbrt_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cc_bmm_bg_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cc_bmm_bg_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ceil_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ceil_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ceil_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_backprop_stats_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_backprop_stats_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_backprop_stats_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_shuffle_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_shuffle_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_shuffle_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_stats_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_stats_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channel_stats_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
channelwise_conv3d_op_cudnn.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
clip_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
clip_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
clip_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
collect_and_distribute_fpn_rpn_proposals_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
collect_and_distribute_fpn_rpn_proposals_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
communicator_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
communicator_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
concat_split_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
concat_split_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
concat_split_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conditional_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conditional_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_cache_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_cache_cudnn.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_cache_cudnn_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_eigen.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_impl.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_shared.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_shared.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_op_shared_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_pool_op_base.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_impl.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_mobile.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_mobile.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_mobile_impl.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_op_mobile_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
conv_transpose_unpool_op_base.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
copy_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
copy_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
copy_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
copy_rows_to_tensor_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
copy_rows_to_tensor_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cos_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cos_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cos_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosh_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosh_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosh_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosine_embedding_criterion_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosine_embedding_criterion_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cosine_embedding_criterion_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
counter_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
counter_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
counter_ops_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
crash_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
create_scope_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
create_scope_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
crf_viterbi_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cross_entropy_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cross_entropy_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cross_entropy_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ctc_beam_search_decoder_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ctc_beam_search_decoder_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ctc_greedy_decoder_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ctc_greedy_decoder_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cube_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cube_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
cube_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
data_couple.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
data_couple.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
data_couple_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dataset_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dataset_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
deform_conv_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
deform_conv_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
deform_conv_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
deform_conv_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
deform_conv_op_impl.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dense_vector_to_id_list_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dense_vector_to_id_list_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
depthwise_3x3_conv_op_cudnn.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
distance_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
distance_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
distance_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
do_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
do_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
do_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dropout_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dropout_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dropout_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
dropout_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_add_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_add_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_add_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_add_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_div_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_div_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_div_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_div_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_linear_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_linear_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_linear_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_logical_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_logical_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_mul_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_mul_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_mul_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_mul_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_op_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_op_test.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops_schema.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops_utils.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_ops_utils.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_sub_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_sub_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_sub_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_sub_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elementwise_sum_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
elu_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
enforce_finite_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
enforce_finite_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
enforce_finite_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ensure_clipped_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ensure_clipped_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ensure_cpu_output_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ensure_cpu_output_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ensure_cpu_output_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
erf_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
erf_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
erf_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
exp_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
exp_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
exp_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_squeeze_dims_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_squeeze_dims_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
expand_squeeze_dims_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fc_inference.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fc_inference.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
feature_maps_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
feature_maps_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
feed_blob_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
feed_blob_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
filler_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
filler_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
filler_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
find_duplicate_elements_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
find_duplicate_elements_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
find_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
find_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
find_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
flatten_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
flatten_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
flexible_top_k.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
flexible_top_k.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
floor_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
floor_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
floor_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
free_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
free_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
free_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fully_connected_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fully_connected_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fully_connected_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_8bit_conversion_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_8bit_conversion_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_nbit_conversion_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_nbit_conversion_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_nbitfake_conversion_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_nbitfake_conversion_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_random_quantization_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
fused_rowwise_random_quantization_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_fused_8bit_rowwise_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_fused_8bit_rowwise_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_op.cuh Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_ranges_to_dense_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gather_ranges_to_dense_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gelu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gelu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gelu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_boxes.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_boxes_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_nms.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_nms_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_nms_gpu.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_nms_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
generate_proposals_op_util_nms_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_byte_string_to_uint8_fill_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_byte_string_to_uint8_fill_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_byte_string_to_uint8_fill_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_fill_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_fill_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
given_tensor_fill_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
glu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
glu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
glu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
group_norm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
group_norm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
group_norm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gru_unit_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gru_unit_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
gru_unit_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
h_softmax_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
h_softmax_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
half_float_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
half_float_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
half_float_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
half_float_ops_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
hard_sigmoid_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
hard_sigmoid_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
hard_sigmoid_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
heatmap_max_keypoint_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
heatmap_max_keypoint_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
histogram_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
histogram_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
if_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
if_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
if_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
im2col_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
im2col_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
im2col_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
index_hash_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
index_hash_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
index_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
index_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
inference_lstm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
inference_lstm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
instance_norm_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
instance_norm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
instance_norm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
instance_norm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
integral_image_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
integral_image_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
integral_image_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
is_empty_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
is_empty_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
jsd_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
jsd_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
key_split_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
key_split_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
last_n_window_collector.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
layer_norm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
layer_norm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
layer_norm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
leaky_relu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
leaky_relu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
leaky_relu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
length_split_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
length_split_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_pad_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_pad_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_pad_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_fused_8bit_rowwise_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_fused_8bit_rowwise_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_fused_nbit_rowwise_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_fused_nbit_rowwise_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_rowwise_8bit_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_reducer_rowwise_8bit_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_tile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_tile_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_tile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_top_k_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lengths_top_k_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
listwise_l2r_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
listwise_l2r_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
load_save_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
load_save_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
load_save_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
load_save_op_util.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
load_save_op_util.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
local_response_normalization_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
local_response_normalization_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
local_response_normalization_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
local_response_normalization_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op_impl.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op_util.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
locally_connected_op_util.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log1p_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log1p_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log1p_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
log_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
logit_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
logit_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
logit_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
loss_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
loss_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
loss_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lp_pool_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lp_pool_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lpnorm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lpnorm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lpnorm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lstm_unit_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lstm_unit_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lstm_unit_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
lstm_utils.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
map_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
map_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
margin_loss_l2r_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
margin_loss_l2r_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
margin_ranking_criterion_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
margin_ranking_criterion_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
margin_ranking_criterion_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
matmul_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
matmul_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
matmul_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
max_pool_with_index.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
max_pool_with_index_gpu.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mean_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mean_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mean_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mem_query_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
merge_id_lists_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
merge_id_lists_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
minmax_gradient_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
minmax_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
minmax_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
minmax_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mish_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mish_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mod_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mod_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
mod_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
moments_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
moments_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
moments_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
multi_class_accuracy_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
multi_class_accuracy_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
multi_class_accuracy_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negate_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negate_gradient_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negate_gradient_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negative_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negative_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
negative_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ngram_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
ngram_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
no_default_engine_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
norm_planar_yuv_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
normalize_l1_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
normalize_l1_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
normalize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
normalize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
normalize_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
numpy_tile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
numpy_tile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
one_hot_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
one_hot_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
one_hot_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
onnx_while_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
onnx_while_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
op_utils_cudnn.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
operator_fallback_gpu.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
operator_fallback_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
order_switch_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
order_switch_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
order_switch_ops_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
order_switch_ops_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pack_rnn_sequence_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pack_rnn_sequence_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pack_segments.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pack_segments.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pack_segments.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pad_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pad_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pad_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
partition_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
partition_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
percentile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
percentile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
perplexity_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
perplexity_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
perplexity_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
piecewise_linear_transform_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
piecewise_linear_transform_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
piecewise_linear_transform_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op_util.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pool_op_util.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pow_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pow_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
pow_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prefetch_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prelu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prelu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prelu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prepend_dim_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prepend_dim_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
prepend_dim_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
quant_decode_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
quant_decode_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
quantile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
quantile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rank_loss_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rank_loss_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reciprocal_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reciprocal_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reciprocal_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reciprocal_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_max_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_max_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_max_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_mean_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_sum_mean_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_sum_mean_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_front_back_sum_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduce_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reducer_functors.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduction_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduction_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reduction_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_n_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_n_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_n_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
relu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
remove_data_blocks_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
remove_data_blocks_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
replace_nan_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
replace_nan_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
replace_nan_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reservoir_sampling.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reshape_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reshape_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reshape_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reshape_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_3d_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_3d_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_3d_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
resize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reverse_packed_segs_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reverse_packed_segs_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
reverse_packed_segs_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rmac_regions_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rmac_regions_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rmac_regions_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rms_norm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rms_norm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rms_norm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_gradient_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_gradient_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_op_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_gradient_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_gradient_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_align_rotated_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_pool_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_pool_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
roi_pool_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rowmul_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rowmul_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rsqrt_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rsqrt_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
rsqrt_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_blobs_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_blobs_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_blobs_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
scale_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
segment_reduction_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
segment_reduction_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
segment_reduction_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
segment_reduction_op_gpu.cuh Forklet of Pytorch 2023-11-08 09:01:59 -07:00
self_binning_histogram_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
self_binning_histogram_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
selu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
selu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
selu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sequence_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sequence_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sequence_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
shape_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
shape_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
shape_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sigmoid_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sigmoid_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sigmoid_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sigmoid_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sigmoid_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sin_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sin_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sin_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sinh_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sinh_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sinh_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sinusoid_position_encoding_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sinusoid_position_encoding_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
slice_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
slice_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
slice_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_utils.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_utils.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_with_loss_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softmax_with_loss_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softplus_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softplus_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softplus_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softsign_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softsign_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
softsign_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
space_batch_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
space_batch_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
space_batch_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_dropout_with_replacement_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_dropout_with_replacement_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_itemwise_dropout_with_replacement_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_itemwise_dropout_with_replacement_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_lp_regularizer_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_lp_regularizer_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_lp_regularizer_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_normalize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_normalize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_normalize_op_gpu.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_to_dense_mask_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_to_dense_mask_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_to_dense_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_to_dense_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sparse_to_dense_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_op_cudnn.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_batch_norm_op_impl.cuh Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_softmax_with_loss_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
spatial_softmax_with_loss_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqr_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqr_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqr_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqrt_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqrt_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
sqrt_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
square_root_divide_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
square_root_divide_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stats_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stats_put_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stats_put_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stop_gradient.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stop_gradient.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stop_gradient_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
string_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
string_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
string_ops_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stump_func_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stump_func_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stump_func_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
stylizer_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
summarize_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
summarize_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
summarize_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
swish_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
swish_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
swish_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tan_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tan_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tan_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tanh_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tanh_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tanh_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tanh_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tanh_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tensor_protos_db_input.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tensor_protos_db_input.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tensor_protos_db_input_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
text_file_reader.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
text_file_reader_utils.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
text_file_reader_utils.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
text_file_reader_utils_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
thresholded_relu_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
thresholded_relu_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
thresholded_relu_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tile_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tile_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tile_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
top_k.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
top_k.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
top_k.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
top_k_heap_selection.cuh Forklet of Pytorch 2023-11-08 09:01:59 -07:00
top_k_radix_selection.cuh Forklet of Pytorch 2023-11-08 09:01:59 -07:00
transpose_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
transpose_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
transpose_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
transpose_op_cudnn.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tt_linear_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
tt_linear_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unique_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unique_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unique_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unsafe_coalesce.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unsafe_coalesce.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
unsafe_coalesce.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
upsample_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
upsample_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
upsample_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
utility_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
utility_ops.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
utility_ops.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
utility_ops_gpu_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
utility_ops_test.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
variable_length_sequence_padding.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
variable_length_sequence_padding.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
weighted_multi_sampling_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
weighted_multi_sampling_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
weighted_sample_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
weighted_sample_op.cu Forklet of Pytorch 2023-11-08 09:01:59 -07:00
weighted_sample_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
while_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
while_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
while_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
workspace_ops.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
zero_gradient_op.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00
zero_gradient_op.h Forklet of Pytorch 2023-11-08 09:01:59 -07:00
zero_gradient_op_gpu.cc Forklet of Pytorch 2023-11-08 09:01:59 -07:00