summaryrefslogtreecommitdiff
path: root/caffe2/operators
ModeNameSize
-rw-r--r--CMakeLists.txt6461logplain
-rw-r--r--abs_op.cc2423logplain
-rw-r--r--abs_op.cu1369logplain
-rw-r--r--abs_op.h705logplain
-rw-r--r--accumulate_op.cc1210logplain
-rw-r--r--accumulate_op.cu194logplain
-rw-r--r--accumulate_op.h1058logplain
-rw-r--r--accuracy_op.cc2521logplain
-rw-r--r--accuracy_op.cu2133logplain
-rw-r--r--accuracy_op.h646logplain
-rw-r--r--acos_op.cc1809logplain
-rw-r--r--acos_op.cu1314logplain
-rw-r--r--acos_op.h711logplain
-rw-r--r--activation_ops_cudnn.h4068logplain
-rw-r--r--adjust_batch_op.cc1025logplain
-rw-r--r--adjust_batch_op.h2503logplain
-rw-r--r--affine_channel_op.cc5398logplain
-rw-r--r--affine_channel_op.cu4358logplain
-rw-r--r--affine_channel_op.h3410logplain
-rw-r--r--apmeter_op.cc3619logplain
-rw-r--r--apmeter_op.h1012logplain
-rw-r--r--arg_ops.cc5879logplain
-rw-r--r--arg_ops.cu2665logplain
-rw-r--r--arg_ops.h2304logplain
-rw-r--r--asin_op.cc1804logplain
-rw-r--r--asin_op.cu1312logplain
-rw-r--r--asin_op.h711logplain
-rw-r--r--assert_op.cc1482logplain
-rw-r--r--assert_op.cu176logplain
-rw-r--r--assert_op.h1320logplain
-rw-r--r--atan_op.cc1802logplain
-rw-r--r--atan_op.cu1305logplain
-rw-r--r--atan_op.h711logplain
-rw-r--r--atomic_ops.cc4759logplain
-rw-r--r--batch_box_cox_op.cc11237logplain
-rw-r--r--batch_box_cox_op.h2172logplain
-rw-r--r--batch_bucketize_op.cc4243logplain
-rw-r--r--batch_bucketize_op.h705logplain
-rw-r--r--batch_gather_ops.cc2035logplain
-rw-r--r--batch_gather_ops.cu4584logplain
-rw-r--r--batch_gather_ops.h4399logplain
-rw-r--r--batch_matmul_op.cc8785logplain
-rw-r--r--batch_matmul_op.cu666logplain
-rw-r--r--batch_matmul_op.h9181logplain
-rw-r--r--batch_matmul_op_gpu_test.cc2535logplain
-rw-r--r--batch_matmul_op_test.cc2214logplain
-rw-r--r--batch_moments_op.cc3603logplain
-rw-r--r--batch_moments_op.cu4112logplain
-rw-r--r--batch_moments_op.h3334logplain
-rw-r--r--batch_sparse_to_dense_op.cc7617logplain
-rw-r--r--batch_sparse_to_dense_op.h1264logplain
-rw-r--r--bbox_transform_op.cc7029logplain
-rw-r--r--bbox_transform_op.h2689logplain
-rw-r--r--bisect_percentile_op.cc3605logplain
-rw-r--r--bisect_percentile_op.h4906logplain
-rw-r--r--boolean_mask_ops.cc16488logplain
-rw-r--r--boolean_mask_ops.cu11692logplain
-rw-r--r--boolean_mask_ops.h1980logplain
-rw-r--r--boolean_unmask_ops.cc4326logplain
-rw-r--r--boolean_unmask_ops.cu3405logplain
-rw-r--r--boolean_unmask_ops.h378logplain
-rw-r--r--boolean_unmask_ops_test.cc1649logplain
-rw-r--r--box_with_nms_limit_op.cc10346logplain
-rw-r--r--box_with_nms_limit_op.h2350logplain
-rw-r--r--byte_weight_dequant_op.cc268logplain
-rw-r--r--byte_weight_dequant_op.h1720logplain
-rw-r--r--cast_op.cc6592logplain
-rw-r--r--cast_op.cu3734logplain
-rw-r--r--cast_op.h1385logplain
-rw-r--r--cbrt_op.cc1816logplain
-rw-r--r--cbrt_op.cu1307logplain
-rw-r--r--cbrt_op.h723logplain
-rw-r--r--ceil_op.cc1612logplain
-rw-r--r--ceil_op.cu739logplain
-rw-r--r--ceil_op.h782logplain
-rw-r--r--channel_backprop_stats_op.cc2987logplain
-rw-r--r--channel_backprop_stats_op.cu6085logplain
-rw-r--r--channel_backprop_stats_op.h722logplain
-rw-r--r--channel_shuffle_op.cc4374logplain
-rw-r--r--channel_shuffle_op.cu6265logplain
-rw-r--r--channel_shuffle_op.h1872logplain
-rw-r--r--channel_stats_op.cc2014logplain
-rw-r--r--channel_stats_op.cu5514logplain
-rw-r--r--channel_stats_op.h696logplain
-rw-r--r--clip_op.cc3342logplain
-rw-r--r--clip_op.cu2228logplain
-rw-r--r--clip_op.h1609logplain
-rw-r--r--collect_and_distribute_fpn_rpn_proposals_op.cc12801logplain
-rw-r--r--collect_and_distribute_fpn_rpn_proposals_op.h3924logplain
-rw-r--r--communicator_op.cc7199logplain
-rw-r--r--communicator_op_gpu.cc774logplain
-rw-r--r--concat_split_op.cc12876logplain
-rw-r--r--concat_split_op.h10839logplain
-rw-r--r--concat_split_op_gpu.cc459logplain
-rw-r--r--conditional_op.cc2180logplain
-rw-r--r--conditional_op.h462logplain
-rw-r--r--conv_gradient_op.cc4854logplain
-rw-r--r--conv_op.cc7516logplain
-rw-r--r--conv_op.h3107logplain
-rw-r--r--conv_op_cache_cudnn.cc419logplain
-rw-r--r--conv_op_cache_cudnn.h1935logplain
-rw-r--r--conv_op_cache_cudnn_test.cc1571logplain
-rw-r--r--conv_op_cudnn.cc50515logplain
-rw-r--r--conv_op_eigen.cc8311logplain
-rw-r--r--conv_op_gpu.cc704logplain
-rw-r--r--conv_op_impl.h28669logplain
-rw-r--r--conv_op_shared.cc1054logplain
-rw-r--r--conv_op_shared.h672logplain
-rw-r--r--conv_op_shared_gpu.cc910logplain
-rw-r--r--conv_pool_op_base.h26426logplain
-rw-r--r--conv_transpose_gradient_op.cc1648logplain
-rw-r--r--conv_transpose_op.cc4418logplain
-rw-r--r--conv_transpose_op.h1697logplain
-rw-r--r--conv_transpose_op_cudnn.cc22161logplain
-rw-r--r--conv_transpose_op_gpu.cc357logplain
-rw-r--r--conv_transpose_op_impl.h17499logplain
-rw-r--r--conv_transpose_op_mobile.cc428logplain
-rw-r--r--conv_transpose_op_mobile.h1517logplain
-rw-r--r--conv_transpose_op_mobile_impl.h19545logplain
-rw-r--r--conv_transpose_op_mobile_test.cc6776logplain
-rw-r--r--conv_transpose_unpool_op_base.h9586logplain
-rw-r--r--copy_op.cc6420logplain
-rw-r--r--copy_op.cu1562logplain
-rw-r--r--copy_op.h1131logplain
-rw-r--r--cos_op.cc2372logplain
-rw-r--r--cos_op.cu1257logplain
-rw-r--r--cos_op.h705logplain
-rw-r--r--cosh_op.cc2410logplain
-rw-r--r--cosh_op.cu1274logplain
-rw-r--r--cosh_op.h711logplain
-rw-r--r--cosine_embedding_criterion_op.cc2570logplain
-rw-r--r--cosine_embedding_criterion_op.cu2085logplain
-rw-r--r--cosine_embedding_criterion_op.h1061logplain
-rw-r--r--counter_ops.cc8886logplain
-rw-r--r--counter_ops.h4507logplain
-rw-r--r--counter_ops_gpu.cc582logplain
-rw-r--r--create_scope_op.cc1241logplain
-rw-r--r--create_scope_op.h5183logplain
-rw-r--r--crf_viterbi_op.cc7274logplain
-rw-r--r--cross_entropy_op.cc22705logplain
-rw-r--r--cross_entropy_op.cu13727logplain
-rw-r--r--cross_entropy_op.h4463logplain
-rw-r--r--ctc_beam_search_decoder_op.cc5739logplain
-rw-r--r--ctc_beam_search_decoder_op.h909logplain
-rw-r--r--ctc_greedy_decoder_op.cc3023logplain
-rw-r--r--ctc_greedy_decoder_op.h802logplain
-rw-r--r--cube_op.cc1723logplain
-rw-r--r--cube_op.cu1266logplain
-rw-r--r--cube_op.h723logplain
-rw-r--r--data_couple.cc590logplain
-rw-r--r--data_couple.h464logplain
-rw-r--r--data_couple_gpu.cu164logplain
-rw-r--r--dataset_ops.cc49961logplain
-rw-r--r--dataset_ops.h5334logplain
-rw-r--r--deform_conv_gradient_op.cc1725logplain
-rw-r--r--deform_conv_op.cc2371logplain
-rw-r--r--deform_conv_op.cu21554logplain
-rw-r--r--deform_conv_op.h3516logplain
-rw-r--r--deform_conv_op_impl.h12820logplain
-rw-r--r--dense_vector_to_id_list_op.cc1182logplain
-rw-r--r--dense_vector_to_id_list_op.h1797logplain
-rw-r--r--depthwise_3x3_conv_op_cudnn.cu17046logplain
-rw-r--r--distance_op.cc22631logplain
-rw-r--r--distance_op.cu14967logplain
-rw-r--r--distance_op.h8061logplain
-rw-r--r--do_op.cc1193logplain
-rw-r--r--do_op.h6909logplain
-rw-r--r--do_op_gpu.cc165logplain
-rw-r--r--dropout_op.cc5597logplain
-rw-r--r--dropout_op.cu2660logplain
-rw-r--r--dropout_op.h1486logplain
-rw-r--r--dropout_op_cudnn.cc8566logplain
-rw-r--r--elementwise_add_gradient_op.cc683logplain
-rw-r--r--elementwise_add_op.cc201logplain
-rw-r--r--elementwise_add_op.h1808logplain
-rw-r--r--elementwise_add_op_gpu.cc394logplain
-rw-r--r--elementwise_div_gradient_op.cc8402logplain
-rw-r--r--elementwise_div_op.cc202logplain
-rw-r--r--elementwise_div_op.cu13989logplain
-rw-r--r--elementwise_div_op.h1062logplain
-rw-r--r--elementwise_linear_op.cc5112logplain
-rw-r--r--elementwise_linear_op.cu3458logplain
-rw-r--r--elementwise_linear_op.h1141logplain
-rw-r--r--elementwise_logical_ops.cc3042logplain
-rw-r--r--elementwise_logical_ops.h5033logplain
-rw-r--r--elementwise_mul_gradient_op.cc3056logplain
-rw-r--r--elementwise_mul_op.cc201logplain
-rw-r--r--elementwise_mul_op.cu5899logplain
-rw-r--r--elementwise_mul_op.h1062logplain
-rw-r--r--elementwise_op_gpu_test.cc975logplain
-rw-r--r--elementwise_op_test.cc764logplain
-rw-r--r--elementwise_op_test.h9237logplain
-rw-r--r--elementwise_ops.cc3747logplain
-rw-r--r--elementwise_ops.cu6720logplain
-rw-r--r--elementwise_ops.h16663logplain
-rw-r--r--elementwise_ops_schema.cc22711logplain
-rw-r--r--elementwise_ops_utils.cc2937logplain
-rw-r--r--elementwise_ops_utils.h812logplain
-rw-r--r--elementwise_sub_gradient_op.cc683logplain
-rw-r--r--elementwise_sub_op.cc201logplain
-rw-r--r--elementwise_sub_op.h1809logplain
-rw-r--r--elementwise_sub_op_gpu.cc394logplain
-rw-r--r--elementwise_sum_op.cc2475logplain
-rw-r--r--elu_op.cc3805logplain
-rw-r--r--elu_op.cu2152logplain
-rw-r--r--elu_op.h875logplain
-rw-r--r--elu_op_cudnn.cc3041logplain
-rw-r--r--enforce_finite_op.cc524logplain
-rw-r--r--enforce_finite_op.cu374logplain
-rw-r--r--enforce_finite_op.h1119logplain
-rw-r--r--ensure_clipped_op.cc1680logplain
-rw-r--r--ensure_clipped_op.h1593logplain
-rw-r--r--ensure_cpu_output_op.cc1139logplain
-rw-r--r--ensure_cpu_output_op.cu293logplain
-rw-r--r--ensure_cpu_output_op.h1442logplain
-rw-r--r--erf_op.cc1803logplain
-rw-r--r--erf_op.cu1326logplain
-rw-r--r--erf_op.h751logplain
-rw-r--r--exp_op.cc1903logplain
-rw-r--r--exp_op.h425logplain
-rw-r--r--exp_op_gpu.cc260logplain
-rw-r--r--expand_op.cc1689logplain
-rw-r--r--expand_op.h3429logplain
-rw-r--r--expand_op_gpu.cc411logplain
-rw-r--r--expand_squeeze_dims_op.cc5867logplain
-rw-r--r--expand_squeeze_dims_op.h3353logplain
-rw-r--r--expand_squeeze_dims_op_gpu.cc252logplain
d---------experimental / c1030logplain
-rw-r--r--fc_inference.cc2148logplain
-rw-r--r--fc_inference.h429logplain
-rw-r--r--feature_maps_ops.cc14276logplain
-rw-r--r--feature_maps_ops.h30007logplain
-rw-r--r--feed_blob_op.cc508logplain
-rw-r--r--feed_blob_op.h769logplain
-rw-r--r--filler_op.cc21408logplain
-rw-r--r--filler_op.cu2156logplain
-rw-r--r--filler_op.h17860logplain
-rw-r--r--find_duplicate_elements_op.cc1509logplain
-rw-r--r--find_duplicate_elements_op.h1556logplain
-rw-r--r--find_op.cc739logplain
-rw-r--r--find_op.cu1483logplain
-rw-r--r--find_op.h2040logplain
-rw-r--r--flatten_op.cc2747logplain
-rw-r--r--flatten_op.h948logplain
-rw-r--r--flexible_top_k.cc5785logplain
-rw-r--r--flexible_top_k.h906logplain
-rw-r--r--floor_op.cc1719logplain
-rw-r--r--floor_op.cu745logplain
-rw-r--r--floor_op.h788logplain
-rw-r--r--free_op.cc408logplain
-rw-r--r--free_op.h738logplain
-rw-r--r--free_op_gpu.cc168logplain
-rw-r--r--fully_connected_op.cc10555logplain
-rw-r--r--fully_connected_op.h9299logplain
-rw-r--r--fully_connected_op_gpu.cc6048logplain
-rw-r--r--fused_rowwise_8bit_conversion_ops.cc5993logplain
-rw-r--r--fused_rowwise_8bit_conversion_ops.h5570logplain
-rw-r--r--fused_rowwise_random_quantization_ops.cc8315logplain
-rw-r--r--fused_rowwise_random_quantization_ops.h2605logplain
-rw-r--r--gather_fused_8bit_rowwise_op.cc1272logplain
-rw-r--r--gather_fused_8bit_rowwise_op.h2179logplain
-rw-r--r--gather_op.cc4174logplain
-rw-r--r--gather_op.cu632logplain
-rw-r--r--gather_op.cuh3738logplain
-rw-r--r--gather_op.h6628logplain
-rw-r--r--gather_ranges_to_dense_op.cc2913logplain
-rw-r--r--gather_ranges_to_dense_op.h4472logplain
-rw-r--r--generate_proposals_op.cc15523logplain
-rw-r--r--generate_proposals_op.cu23427logplain
-rw-r--r--generate_proposals_op.h6271logplain
-rw-r--r--generate_proposals_op_gpu_test.cc30026logplain
-rw-r--r--generate_proposals_op_test.cc33201logplain
-rw-r--r--generate_proposals_op_util_boxes.h13690logplain
-rw-r--r--generate_proposals_op_util_boxes_test.cc5336logplain
-rw-r--r--generate_proposals_op_util_nms.h24758logplain
-rw-r--r--generate_proposals_op_util_nms_gpu.cu18595logplain
-rw-r--r--generate_proposals_op_util_nms_gpu.h2067logplain
-rw-r--r--generate_proposals_op_util_nms_gpu_test.cc21584logplain
-rw-r--r--generate_proposals_op_util_nms_test.cc13377logplain
-rw-r--r--given_tensor_byte_string_to_uint8_fill_op.cc2299logplain
-rw-r--r--given_tensor_byte_string_to_uint8_fill_op.cu248logplain
-rw-r--r--given_tensor_byte_string_to_uint8_fill_op.h2059logplain
-rw-r--r--given_tensor_fill_op.cc7402logplain
-rw-r--r--given_tensor_fill_op.cu556logplain
-rw-r--r--given_tensor_fill_op.h2914logplain
-rw-r--r--glu_op.cc1474logplain
-rw-r--r--glu_op.cu1142logplain
-rw-r--r--glu_op.h1443logplain
-rw-r--r--group_norm_op.cc6765logplain
-rw-r--r--group_norm_op.cu15141logplain
-rw-r--r--group_norm_op.h8298logplain
-rw-r--r--gru_unit_op.cc1894logplain
-rw-r--r--gru_unit_op.h6644logplain
-rw-r--r--gru_unit_op_gpu.cu3572logplain
-rw-r--r--h_softmax_op.cc23217logplain
-rw-r--r--h_softmax_op.h4928logplain
-rw-r--r--half_float_ops.cc1858logplain
-rw-r--r--half_float_ops.cu1413logplain
-rw-r--r--half_float_ops.h2482logplain
-rw-r--r--hard_sigmoid_op.cc3852logplain
-rw-r--r--hard_sigmoid_op.cu2181logplain
-rw-r--r--hard_sigmoid_op.h994logplain
-rw-r--r--heatmap_max_keypoint_op.cc6048logplain
-rw-r--r--heatmap_max_keypoint_op.h818logplain
d---------hip515logplain
-rw-r--r--if_op.cc734logplain
-rw-r--r--if_op.h1755logplain
-rw-r--r--if_op_gpu.cc165logplain
-rw-r--r--im2col_op.cc3835logplain
-rw-r--r--im2col_op.h8913logplain
-rw-r--r--im2col_op_gpu.cc245logplain
-rw-r--r--index_hash_ops.cc1152logplain
-rw-r--r--index_hash_ops.h2103logplain
-rw-r--r--index_ops.cc11441logplain
-rw-r--r--index_ops.h3043logplain
-rw-r--r--instance_norm_gradient_op.cc9992logplain
-rw-r--r--instance_norm_op.cc7120logplain
-rw-r--r--instance_norm_op.cu16578logplain
-rw-r--r--instance_norm_op.h2676logplain
-rw-r--r--integral_image_op.cc5031logplain
-rw-r--r--integral_image_op.cu6808logplain
-rw-r--r--integral_image_op.h875logplain
-rw-r--r--is_empty_op.cc1557logplain
-rw-r--r--is_empty_op.h558logplain
-rw-r--r--jsd_op.cc2848logplain
-rw-r--r--jsd_op.h721logplain
-rw-r--r--key_split_ops.cc308logplain
-rw-r--r--key_split_ops.h1385logplain
-rw-r--r--last_n_window_collector.cc6018logplain
-rw-r--r--layer_norm_op.cc6382logplain
-rw-r--r--layer_norm_op.cu6748logplain
-rw-r--r--layer_norm_op.h6233logplain
-rw-r--r--leaky_relu_op.cc3403logplain
-rw-r--r--leaky_relu_op.cu1667logplain
-rw-r--r--leaky_relu_op.h1079logplain
-rw-r--r--length_split_op.cc1194logplain
-rw-r--r--length_split_op.h2244logplain
-rw-r--r--lengths_pad_op.cc1861logplain
-rw-r--r--lengths_pad_op.cu188logplain
-rw-r--r--lengths_pad_op.h2559logplain
-rw-r--r--lengths_reducer_fused_8bit_rowwise_ops.cc3974logplain
-rw-r--r--lengths_reducer_fused_8bit_rowwise_ops.h2719logplain
-rw-r--r--lengths_reducer_ops.cc5185logplain
-rw-r--r--lengths_reducer_ops.h3508logplain
-rw-r--r--lengths_reducer_rowwise_8bit_ops.cc8053logplain
-rw-r--r--lengths_reducer_rowwise_8bit_ops.h6159logplain
-rw-r--r--lengths_tile_op.cc2821logplain
-rw-r--r--lengths_tile_op.cu3299logplain
-rw-r--r--lengths_tile_op.h582logplain
-rw-r--r--lengths_top_k_op.cc5233logplain
-rw-r--r--lengths_top_k_op.h1294logplain
-rw-r--r--listwise_l2r_op.cc8594logplain
-rw-r--r--listwise_l2r_op.h1500logplain
-rw-r--r--load_save_op.cc8888logplain
-rw-r--r--load_save_op.h19407logplain
-rw-r--r--load_save_op_gpu.cc632logplain
-rw-r--r--local_response_normalization_op.cc15160logplain
-rw-r--r--local_response_normalization_op.cu11528logplain
-rw-r--r--local_response_normalization_op.h2759logplain
-rw-r--r--local_response_normalization_op_cudnn.cc5756logplain
-rw-r--r--locally_connected_op.cc5364logplain
-rw-r--r--locally_connected_op.h3842logplain
-rw-r--r--locally_connected_op_gpu.cc847logplain
-rw-r--r--locally_connected_op_impl.h26495logplain
-rw-r--r--locally_connected_op_util.cc2108logplain
-rw-r--r--locally_connected_op_util.h1334logplain
-rw-r--r--log_op.cc1936logplain
-rw-r--r--log_op.h431logplain
-rw-r--r--log_op_gpu.cc260logplain
-rw-r--r--logit_op.cc2571logplain
-rw-r--r--logit_op.cu1668logplain
-rw-r--r--logit_op.h1021logplain
-rw-r--r--loss_op.cc1844logplain
-rw-r--r--loss_op.cu287logplain
-rw-r--r--loss_op.h1028logplain
-rw-r--r--lp_pool_op.cc10733logplain
-rw-r--r--lp_pool_op.cu10498logplain
-rw-r--r--lpnorm_op.cc5178logplain
-rw-r--r--lpnorm_op.h1213logplain
-rw-r--r--lstm_unit_op.cc1811logplain
-rw-r--r--lstm_unit_op.h8897logplain
-rw-r--r--lstm_unit_op_gpu.cu7147logplain
-rw-r--r--map_ops.cc2494logplain
-rw-r--r--map_ops.h7935logplain
-rw-r--r--margin_ranking_criterion_op.cc3521logplain
-rw-r--r--margin_ranking_criterion_op.cu2633logplain
-rw-r--r--margin_ranking_criterion_op.h1047logplain
-rw-r--r--matmul_op.cc7097logplain
-rw-r--r--matmul_op.h2828logplain
-rw-r--r--matmul_op_gpu.cc164logplain
-rw-r--r--max_pool_with_index.cu7722logplain
-rw-r--r--max_pool_with_index_gpu.h1155logplain
-rw-r--r--mean_op.cc2548logplain
-rw-r--r--mean_op.cu237logplain
-rw-r--r--mean_op.h3021logplain
-rw-r--r--mem_query_op.cu1527logplain
-rw-r--r--merge_id_lists_op.cc1430logplain
-rw-r--r--merge_id_lists_op.h2416logplain
-rw-r--r--minmax_gradient_ops.cc2107logplain
-rw-r--r--minmax_ops.cc3717logplain
-rw-r--r--minmax_ops.cu1586logplain
-rw-r--r--minmax_ops.h3799logplain
-rw-r--r--mod_op.cc2276logplain
-rw-r--r--mod_op.h969logplain
-rw-r--r--moments_op.cc2600logplain
-rw-r--r--moments_op.cu3080logplain
-rw-r--r--moments_op.h3726logplain
-rw-r--r--multi_class_accuracy_op.cc2295logplain
-rw-r--r--multi_class_accuracy_op.cu2175logplain
-rw-r--r--multi_class_accuracy_op.h539logplain
-rw-r--r--negate_gradient_op.cc802logplain
-rw-r--r--negate_gradient_op.h566logplain
-rw-r--r--negate_gradient_op_gpu.cc198logplain
-rw-r--r--negative_op.cc1709logplain
-rw-r--r--negative_op.h451logplain
-rw-r--r--negative_op_gpu.cc269logplain
-rw-r--r--ngram_ops.cc357logplain
-rw-r--r--ngram_ops.h2629logplain
-rw-r--r--no_default_engine_op.h1063logplain
-rw-r--r--norm_planar_yuv_op.cc1566logplain
-rw-r--r--normalize_l1_op.cc1156logplain
-rw-r--r--normalize_l1_op.h1075logplain
-rw-r--r--normalize_op.cc2778logplain
-rw-r--r--normalize_op.h2283logplain
-rw-r--r--normalize_ops.cu4972logplain
-rw-r--r--numpy_tile_op.cc503logplain
-rw-r--r--numpy_tile_op.h3576logplain
-rw-r--r--one_hot_ops.cc11896logplain
-rw-r--r--one_hot_ops.cu929logplain
-rw-r--r--one_hot_ops.h2411logplain
-rw-r--r--onnx_while_op.cc3351logplain
-rw-r--r--onnx_while_op.h10540logplain
-rw-r--r--onnxifi_op.cc8007logplain
-rw-r--r--onnxifi_op.h8534logplain
-rw-r--r--op_utils_cudnn.h2112logplain
-rw-r--r--operator_fallback_gpu.h4146logplain
-rw-r--r--operator_fallback_gpu_test.cc2538logplain
-rw-r--r--order_switch_ops.cc3120logplain
-rw-r--r--order_switch_ops.h2149logplain
-rw-r--r--order_switch_ops_cudnn.cc5817logplain
-rw-r--r--order_switch_ops_gpu.cc265logplain
-rw-r--r--pack_rnn_sequence_op.cc3476logplain
-rw-r--r--pack_rnn_sequence_op.h3059logplain
-rw-r--r--pack_segments.cc6981logplain
-rw-r--r--pack_segments.cu9818logplain
-rw-r--r--pack_segments.h2547logplain
-rw-r--r--pad_op.cc15401logplain
-rw-r--r--pad_op.h2873logplain
-rw-r--r--pad_op_gpu.cu16566logplain
-rw-r--r--partition_ops.cc4171logplain
-rw-r--r--partition_ops.h9898logplain
-rw-r--r--percentile_op.cc5949logplain
-rw-r--r--percentile_op.h895logplain
-rw-r--r--perplexity_op.cc1259logplain
-rw-r--r--perplexity_op.cu1190logplain
-rw-r--r--perplexity_op.h447logplain
-rw-r--r--piecewise_linear_transform_op.cc3491logplain
-rw-r--r--piecewise_linear_transform_op.cu8023logplain
-rw-r--r--piecewise_linear_transform_op.h8153logplain
-rw-r--r--pool_gradient_op.cc26669logplain
-rw-r--r--pool_op.cc42490logplain
-rw-r--r--pool_op.cu60182logplain
-rw-r--r--pool_op.h8495logplain
-rw-r--r--pool_op_cudnn.cc17711logplain
-rw-r--r--pool_op_util.cc11285logplain
-rw-r--r--pool_op_util.h1105logplain
-rw-r--r--pow_op.cc12444logplain
-rw-r--r--pow_op.cu2660logplain
-rw-r--r--pow_op.h4664logplain
-rw-r--r--prefetch_op.h4655logplain
-rw-r--r--prelu_op.cc10257logplain
-rw-r--r--prelu_op.cu7428logplain
-rw-r--r--prelu_op.h1037logplain
-rw-r--r--prepend_dim_op.cc1308logplain
-rw-r--r--prepend_dim_op.h2624logplain
-rw-r--r--prepend_dim_op_gpu.cc248logplain
-rw-r--r--quant_decode_op.cc2147logplain
-rw-r--r--quant_decode_op.h5340logplain
d---------quantized2228logplain
-rw-r--r--rank_loss_op.cc6217logplain
-rw-r--r--rank_loss_op.h820logplain
-rw-r--r--reciprocal_gradient_op.cc1311logplain
-rw-r--r--reciprocal_op.cc1345logplain
-rw-r--r--reciprocal_op.cu1341logplain
-rw-r--r--reciprocal_op.h721logplain
-rw-r--r--reduce_front_back_max_ops.cc8403logplain
-rw-r--r--reduce_front_back_max_ops.cu4715logplain
-rw-r--r--reduce_front_back_max_ops.h4377logplain
-rw-r--r--reduce_front_back_mean_ops.cc8526logplain
-rw-r--r--reduce_front_back_sum_mean_ops.cu7047logplain
-rw-r--r--reduce_front_back_sum_mean_ops.h5315logplain
-rw-r--r--reduce_front_back_sum_ops.cc8300logplain
-rw-r--r--reduce_ops.cc14010logplain
-rw-r--r--reduce_ops.cu4508logplain
-rw-r--r--reduce_ops.h8457logplain
-rw-r--r--reducer_functors.h24551logplain
-rw-r--r--reduction_ops.cc9524logplain
-rw-r--r--reduction_ops.cu3736logplain
-rw-r--r--reduction_ops.h4720logplain
-rw-r--r--relu_n_op.cc2785logplain
-rw-r--r--relu_n_op.cu2057logplain
-rw-r--r--relu_n_op.h990logplain
-rw-r--r--relu_op.cc3958logplain
-rw-r--r--relu_op.cu5069logplain
-rw-r--r--relu_op.h624logplain
-rw-r--r--relu_op_cudnn.cc305logplain
-rw-r--r--remove_data_blocks_op.cc798logplain
-rw-r--r--remove_data_blocks_op.h2651logplain
-rw-r--r--replace_nan_op.cc854logplain
-rw-r--r--replace_nan_op.cu745logplain
-rw-r--r--replace_nan_op.h1155logplain
-rw-r--r--reservoir_sampling.cc9048logplain
-rw-r--r--reshape_op.cc5796logplain
-rw-r--r--reshape_op.h4227logplain
-rw-r--r--reshape_op_gpu.cc217logplain
-rw-r--r--reshape_op_gpu_test.cc1457logplain
-rw-r--r--resize_op.cc11195logplain
-rw-r--r--resize_op.cu4759logplain
-rw-r--r--resize_op.h2176logplain
-rw-r--r--reverse_packed_segs_op.cc1119logplain
-rw-r--r--reverse_packed_segs_op.cu2530logplain
-rw-r--r--reverse_packed_segs_op.h2772logplain
-rw-r--r--rmac_regions_op.cc3643logplain
-rw-r--r--rmac_regions_op.cu5872logplain
-rw-r--r--rmac_regions_op.h693logplain
d---------rnn815logplain
-rw-r--r--roi_align_gradient_op.cc7602logplain
-rw-r--r--roi_align_gradient_op.cu7021logplain
-rw-r--r--roi_align_gradient_op.h1178logplain
-rw-r--r--roi_align_op.cc11677logplain
-rw-r--r--roi_align_op.cu5395logplain
-rw-r--r--roi_align_op.h1387logplain
-rw-r--r--roi_align_op_gpu_test.cc8972logplain
-rw-r--r--roi_align_rotated_gradient_op.cc924logplain
-rw-r--r--roi_align_rotated_gradient_op.cu6969logplain
-rw-r--r--roi_align_rotated_gradient_op.h1243logplain
-rw-r--r--roi_align_rotated_op.cc12286logplain
-rw-r--r--roi_align_rotated_op.cu5681logplain
-rw-r--r--roi_align_rotated_op.h1425logplain
-rw-r--r--roi_pool_op.cc7358logplain
-rw-r--r--roi_pool_op.cu6904logplain
-rw-r--r--roi_pool_op.h2442logplain
-rw-r--r--rowmul_op.cc1550logplain
-rw-r--r--rowmul_op.h1947logplain
-rw-r--r--rsqrt_op.cc1758logplain
-rw-r--r--rsqrt_op.cu1410logplain
-rw-r--r--rsqrt_op.h729logplain
-rw-r--r--scale_op.cc852logplain
-rw-r--r--scale_op.h996logplain
-rw-r--r--scale_op_gpu.cc309logplain
-rw-r--r--segment_reduction_op.cc20662logplain
-rw-r--r--segment_reduction_op.h70493logplain
-rw-r--r--segment_reduction_op_gpu.cu57134logplain
-rw-r--r--selu_op.cc4131logplain
-rw-r--r--selu_op.cu1884logplain
-rw-r--r--selu_op.h1515logplain
-rw-r--r--sequence_ops.cc15958logplain
-rw-r--r--sequence_ops.cu10848logplain
-rw-r--r--sequence_ops.h8204logplain
-rw-r--r--shape_op.cc1769logplain
-rw-r--r--shape_op.h1623logplain
-rw-r--r--shape_op_gpu.cc151logplain
-rw-r--r--sigmoid_gradient_op.cc1297logplain
-rw-r--r--sigmoid_op.cc1960logplain
-rw-r--r--sigmoid_op.cu1977logplain
-rw-r--r--sigmoid_op.h639logplain
-rw-r--r--sigmoid_op_cudnn.cc320logplain
-rw-r--r--sin_op.cc2346logplain
-rw-r--r--sin_op.cu1267logplain
-rw-r--r--sin_op.h705logplain
-rw-r--r--sinh_op.cc2406logplain
-rw-r--r--sinh_op.cu1274logplain
-rw-r--r--sinh_op.h711logplain
-rw-r--r--sinusoid_position_encoding_op.cc1404logplain
-rw-r--r--sinusoid_position_encoding_op.h2811logplain
-rw-r--r--slice_op.cc3720logplain
-rw-r--r--slice_op.cu10822logplain
-rw-r--r--slice_op.h9974logplain
-rw-r--r--softmax_op.cc6327logplain
-rw-r--r--softmax_op.h1141logplain
-rw-r--r--softmax_op_cudnn.cc3966logplain
-rw-r--r--softmax_ops.cu23598logplain
-rw-r--r--softmax_shared.cc1375logplain
-rw-r--r--softmax_shared.h462logplain
-rw-r--r--softmax_with_loss_op.cc12895logplain
-rw-r--r--softmax_with_loss_op.h2525logplain
-rw-r--r--softplus_op.cc2977logplain
-rw-r--r--softplus_op.cu1609logplain
-rw-r--r--softplus_op.h781logplain
-rw-r--r--softsign_op.cc3623logplain
-rw-r--r--softsign_op.cu2031logplain
-rw-r--r--softsign_op.h675logplain
-rw-r--r--space_batch_op.cc3909logplain
-rw-r--r--space_batch_op.h6833logplain
-rw-r--r--space_batch_op_gpu.cu4852logplain
-rw-r--r--sparse_normalize_op.cc2535logplain
-rw-r--r--sparse_normalize_op.h826logplain
-rw-r--r--sparse_normalize_op_gpu.cu226logplain
-rw-r--r--sparse_to_dense_mask_op.cc4261logplain
-rw-r--r--sparse_to_dense_mask_op.h9592logplain
-rw-r--r--sparse_to_dense_op.cc1914logplain
-rw-r--r--sparse_to_dense_op.cu2387logplain
-rw-r--r--sparse_to_dense_op.h3962logplain
-rw-r--r--spatial_batch_norm_gradient_op.cc6326logplain
-rw-r--r--spatial_batch_norm_op.cc6003logplain
-rw-r--r--spatial_batch_norm_op.cu294logplain
-rw-r--r--spatial_batch_norm_op.h14850logplain
-rw-r--r--spatial_batch_norm_op_cudnn.cu12449logplain
-rw-r--r--spatial_batch_norm_op_impl.cuh16338logplain
-rw-r--r--spatial_softmax_with_loss_op.cc8373logplain
-rw-r--r--spatial_softmax_with_loss_op.h2134logplain
-rw-r--r--sqr_op.cc2107logplain
-rw-r--r--sqr_op.h431logplain
-rw-r--r--sqr_op_gpu.cc260logplain
-rw-r--r--sqrt_op.cc2226logplain
-rw-r--r--sqrt_op.h448logplain
-rw-r--r--sqrt_op_gpu.cc263logplain
-rw-r--r--square_root_divide_op.cc1083logplain
-rw-r--r--square_root_divide_op.h1788logplain
-rw-r--r--stats_ops.cc9858logplain
-rw-r--r--stats_put_ops.cc3630logplain
-rw-r--r--stats_put_ops.h2748logplain
-rw-r--r--stop_gradient.cc551logplain
-rw-r--r--stop_gradient.h548logplain
-rw-r--r--stop_gradient_gpu.cc191logplain
-rw-r--r--string_ops.cc6210logplain
-rw-r--r--string_ops.h2051logplain
-rw-r--r--string_ops_test.cc4087logplain
-rw-r--r--stump_func_op.cc2724logplain
-rw-r--r--stump_func_op.cu1589logplain
-rw-r--r--stump_func_op.h2082logplain
-rw-r--r--stylizer_ops.cc19473logplain
-rw-r--r--summarize_op.cc2285logplain
-rw-r--r--summarize_op.cu3385logplain
-rw-r--r--summarize_op.h1866logplain
-rw-r--r--swish_op.cc2662logplain
-rw-r--r--swish_op.cu2128logplain
-rw-r--r--swish_op.h772logplain
-rw-r--r--tan_op.cc1742logplain
-rw-r--r--tan_op.cu1345logplain
-rw-r--r--tan_op.h705logplain
-rw-r--r--tanh_gradient_op.cc1286logplain
-rw-r--r--tanh_op.cc1848logplain
-rw-r--r--tanh_op.cu1305logplain
-rw-r--r--tanh_op.h723logplain
-rw-r--r--tanh_op_cudnn.cc305logplain
-rw-r--r--tensor_protos_db_input.cc1701logplain
-rw-r--r--tensor_protos_db_input.h3633logplain
-rw-r--r--tensor_protos_db_input_gpu.cc248logplain
-rw-r--r--text_file_reader.cc6351logplain
-rw-r--r--text_file_reader_utils.cc3093logplain
-rw-r--r--text_file_reader_utils.h2907logplain
-rw-r--r--text_file_reader_utils_test.cc4243logplain
-rw-r--r--thresholded_relu_op.cc2902logplain
-rw-r--r--thresholded_relu_op.cu1642logplain
-rw-r--r--thresholded_relu_op.h1107logplain
-rw-r--r--tile_op.cc3423logplain
-rw-r--r--tile_op.cu2507logplain
-rw-r--r--tile_op.h8143logplain
-rw-r--r--top_k.cc9924logplain
-rw-r--r--top_k.cu10794logplain
-rw-r--r--top_k.h1086logplain
-rw-r--r--top_k_heap_selection.cuh8923logplain
-rw-r--r--top_k_radix_selection.cuh15557logplain
-rw-r--r--transpose_op.cc3434logplain
-rw-r--r--transpose_op.cu185logplain
-rw-r--r--transpose_op.h1977logplain
-rw-r--r--transpose_op_cudnn.cc4259logplain
-rw-r--r--tt_linear_op.cc2569logplain
-rw-r--r--tt_linear_op.h6418logplain
-rw-r--r--unique_ops.cc3534logplain
-rw-r--r--unique_ops.cu4252logplain
-rw-r--r--unique_ops.h1666logplain
-rw-r--r--upsample_op.cc6978logplain
-rw-r--r--upsample_op.cu7926logplain
-rw-r--r--upsample_op.h2184logplain
-rw-r--r--utility_ops.cc30832logplain
-rw-r--r--utility_ops.cu9271logplain
-rw-r--r--utility_ops.h41918logplain
-rw-r--r--utility_ops_cudnn.cc5424logplain
-rw-r--r--utility_ops_gpu_test.cc1427logplain
-rw-r--r--utility_ops_test.cc1223logplain
-rw-r--r--variable_length_sequence_padding.cc713logplain
-rw-r--r--variable_length_sequence_padding.h1376logplain
-rw-r--r--weighted_multi_sampling_op.cc4034logplain
-rw-r--r--weighted_multi_sampling_op.h587logplain
-rw-r--r--weighted_sample_op.cc4109logplain
-rw-r--r--weighted_sample_op.cu2842logplain
-rw-r--r--weighted_sample_op.h724logplain
-rw-r--r--while_op.cc841logplain
-rw-r--r--while_op.h1952logplain
-rw-r--r--while_op_gpu.cc174logplain
-rw-r--r--workspace_ops.cc1258logplain
-rw-r--r--zero_gradient_op.cc797logplain
-rw-r--r--zero_gradient_op.h347logplain
-rw-r--r--zero_gradient_op_gpu.cc173logplain