package info
(click to toggle)
Folder: operators
 |
.. (parent) | |||
|
d | rwxr-xr-x | 4,096 | hip |
|
d | rwxr-xr-x | 4,096 | quantized |
|
d | rwxr-xr-x | 4,096 | rnn |
 |
- | rw-r--r-- | 3,097 | CMakeLists.txt |
|
- | rw-r--r-- | 2,446 | abs_op.cc |
|
- | rw-r--r-- | 1,369 | abs_op.cu |
|
- | rw-r--r-- | 705 | abs_op.h |
|
- | rw-r--r-- | 1,210 | accumulate_op.cc |
|
- | rw-r--r-- | 194 | accumulate_op.cu |
|
- | rw-r--r-- | 1,073 | accumulate_op.h |
|
- | rw-r--r-- | 2,521 | accuracy_op.cc |
|
- | rw-r--r-- | 2,133 | accuracy_op.cu |
|
- | rw-r--r-- | 652 | accuracy_op.h |
|
- | rw-r--r-- | 1,809 | acos_op.cc |
|
- | rw-r--r-- | 1,314 | acos_op.cu |
|
- | rw-r--r-- | 711 | acos_op.h |
|
- | rw-r--r-- | 4,122 | activation_ops_cudnn.h |
|
- | rw-r--r-- | 5,398 | affine_channel_op.cc |
|
- | rw-r--r-- | 4,358 | affine_channel_op.cu |
|
- | rw-r--r-- | 3,450 | affine_channel_op.h |
|
- | rw-r--r-- | 840 | alias_with_name.cc |
|
- | rw-r--r-- | 301 | alias_with_name.cu |
|
- | rw-r--r-- | 1,234 | alias_with_name.h |
|
- | rw-r--r-- | 3,619 | apmeter_op.cc |
|
- | rw-r--r-- | 1,027 | apmeter_op.h |
|
- | rw-r--r-- | 5,879 | arg_ops.cc |
|
- | rw-r--r-- | 2,665 | arg_ops.cu |
|
- | rw-r--r-- | 2,319 | arg_ops.h |
|
- | rw-r--r-- | 1,804 | asin_op.cc |
|
- | rw-r--r-- | 1,312 | asin_op.cu |
|
- | rw-r--r-- | 711 | asin_op.h |
|
- | rw-r--r-- | 1,488 | assert_op.cc |
|
- | rw-r--r-- | 176 | assert_op.cu |
|
- | rw-r--r-- | 1,335 | assert_op.h |
|
- | rw-r--r-- | 1,802 | atan_op.cc |
|
- | rw-r--r-- | 1,305 | atan_op.cu |
|
- | rw-r--r-- | 711 | atan_op.h |
|
- | rw-r--r-- | 5,672 | atomic_ops.cc |
|
- | rw-r--r-- | 11,453 | batch_box_cox_op.cc |
|
- | rw-r--r-- | 2,287 | batch_box_cox_op.h |
|
- | rw-r--r-- | 4,243 | batch_bucketize_op.cc |
|
- | rw-r--r-- | 720 | batch_bucketize_op.h |
|
- | rw-r--r-- | 2,042 | batch_gather_ops.cc |
|
- | rw-r--r-- | 4,691 | batch_gather_ops.cu |
|
- | rw-r--r-- | 5,264 | batch_gather_ops.h |
|
- | rw-r--r-- | 8,795 | batch_matmul_op.cc |
|
- | rw-r--r-- | 672 | batch_matmul_op.cu |
|
- | rw-r--r-- | 9,602 | batch_matmul_op.h |
|
- | rw-r--r-- | 2,535 | batch_matmul_op_gpu_test.cc |
|
- | rw-r--r-- | 2,214 | batch_matmul_op_test.cc |
|
- | rw-r--r-- | 3,603 | batch_moments_op.cc |
|
- | rw-r--r-- | 4,112 | batch_moments_op.cu |
|
- | rw-r--r-- | 3,364 | batch_moments_op.h |
|
- | rw-r--r-- | 4,506 | batch_permutation_op.cc |
|
- | rw-r--r-- | 2,947 | batch_permutation_op.cu |
|
- | rw-r--r-- | 954 | batch_permutation_op.h |
|
- | rw-r--r-- | 7,903 | batch_permutation_op_gpu_test.cc |
|
- | rw-r--r-- | 6,128 | batch_sparse_to_dense_op.cc |
|
- | rw-r--r-- | 4,065 | batch_sparse_to_dense_op.cu |
|
- | rw-r--r-- | 4,578 | batch_sparse_to_dense_op.h |
|
- | rw-r--r-- | 7,905 | bbox_transform_op.cc |
|
- | rw-r--r-- | 2,668 | bbox_transform_op.h |
|
- | rw-r--r-- | 3,606 | bisect_percentile_op.cc |
|
- | rw-r--r-- | 4,921 | bisect_percentile_op.h |
|
- | rw-r--r-- | 17,760 | boolean_mask_ops.cc |
|
- | rw-r--r-- | 11,719 | boolean_mask_ops.cu |
|
- | rw-r--r-- | 2,665 | boolean_mask_ops.h |
|
- | rw-r--r-- | 4,326 | boolean_unmask_ops.cc |
|
- | rw-r--r-- | 3,437 | boolean_unmask_ops.cu |
|
- | rw-r--r-- | 378 | boolean_unmask_ops.h |
|
- | rw-r--r-- | 1,649 | boolean_unmask_ops_test.cc |
|
- | rw-r--r-- | 12,256 | box_with_nms_limit_op.cc |
|
- | rw-r--r-- | 4,747 | box_with_nms_limit_op.h |
|
- | rw-r--r-- | 2,324 | bucketize_op.cc |
|
- | rw-r--r-- | 1,393 | bucketize_op.cu |
|
- | rw-r--r-- | 1,361 | bucketize_op.h |
|
- | rw-r--r-- | 268 | byte_weight_dequant_op.cc |
|
- | rw-r--r-- | 1,722 | byte_weight_dequant_op.h |
|
- | rw-r--r-- | 6,592 | cast_op.cc |
|
- | rw-r--r-- | 3,734 | cast_op.cu |
|
- | rw-r--r-- | 1,393 | cast_op.h |
|
- | rw-r--r-- | 1,824 | cbrt_op.cc |
|
- | rw-r--r-- | 1,307 | cbrt_op.cu |
|
- | rw-r--r-- | 723 | cbrt_op.h |
|
- | rw-r--r-- | 274 | cc_bmm_bg_op.cc |
|
- | rw-r--r-- | 3,894 | cc_bmm_bg_op.h |
|
- | rw-r--r-- | 1,612 | ceil_op.cc |
|
- | rw-r--r-- | 739 | ceil_op.cu |
|
- | rw-r--r-- | 782 | ceil_op.h |
|
- | rw-r--r-- | 2,987 | channel_backprop_stats_op.cc |
|
- | rw-r--r-- | 6,085 | channel_backprop_stats_op.cu |
|
- | rw-r--r-- | 737 | channel_backprop_stats_op.h |
|
- | rw-r--r-- | 4,437 | channel_shuffle_op.cc |
|
- | rw-r--r-- | 6,329 | channel_shuffle_op.cu |
|
- | rw-r--r-- | 1,902 | channel_shuffle_op.h |
|
- | rw-r--r-- | 2,196 | channel_stats_op.cc |
|
- | rw-r--r-- | 2,977 | channel_stats_op.cu |
|
- | rw-r--r-- | 1,807 | channel_stats_op.h |
|
- | rw-r--r-- | 21,183 | channelwise_conv3d_op_cudnn.cu |
|
- | rw-r--r-- | 3,342 | clip_op.cc |
|
- | rw-r--r-- | 2,228 | clip_op.cu |
|
- | rw-r--r-- | 1,639 | clip_op.h |
|
- | rw-r--r-- | 23,171 | collect_and_distribute_fpn_rpn_proposals_op.cc |
|
- | rw-r--r-- | 6,875 | collect_and_distribute_fpn_rpn_proposals_op.h |
|
- | rw-r--r-- | 7,199 | communicator_op.cc |
|
- | rw-r--r-- | 774 | communicator_op_gpu.cc |
|
- | rw-r--r-- | 17,622 | concat_split_op.cc |
|
- | rw-r--r-- | 11,721 | concat_split_op.h |
|
- | rw-r--r-- | 459 | concat_split_op_gpu.cc |
|
- | rw-r--r-- | 2,180 | conditional_op.cc |
|
- | rw-r--r-- | 487 | conditional_op.h |
|
- | rw-r--r-- | 4,856 | conv_gradient_op.cc |
|
- | rw-r--r-- | 7,516 | conv_op.cc |
|
- | rw-r--r-- | 3,125 | conv_op.h |
|
- | rw-r--r-- | 419 | conv_op_cache_cudnn.cc |
|
- | rw-r--r-- | 1,935 | conv_op_cache_cudnn.h |
|
- | rw-r--r-- | 1,571 | conv_op_cache_cudnn_test.cc |
|
- | rw-r--r-- | 51,857 | conv_op_cudnn.cc |
|
- | rw-r--r-- | 8,329 | conv_op_eigen.cc |
|
- | rw-r--r-- | 704 | conv_op_gpu.cc |
|
- | rw-r--r-- | 28,678 | conv_op_impl.h |
|
- | rw-r--r-- | 1,054 | conv_op_shared.cc |
|
- | rw-r--r-- | 672 | conv_op_shared.h |
|
- | rw-r--r-- | 910 | conv_op_shared_gpu.cc |
|
- | rw-r--r-- | 31,706 | conv_pool_op_base.h |
|
- | rw-r--r-- | 1,648 | conv_transpose_gradient_op.cc |
|
- | rw-r--r-- | 4,418 | conv_transpose_op.cc |
|
- | rw-r--r-- | 1,727 | conv_transpose_op.h |
|
- | rw-r--r-- | 28,679 | conv_transpose_op_cudnn.cc |
|
- | rw-r--r-- | 357 | conv_transpose_op_gpu.cc |
|
- | rw-r--r-- | 18,264 | conv_transpose_op_impl.h |
|
- | rw-r--r-- | 364 | conv_transpose_op_mobile.cc |
|
- | rw-r--r-- | 1,470 | conv_transpose_op_mobile.h |
|
- | rw-r--r-- | 19,587 | conv_transpose_op_mobile_impl.h |
|
- | rw-r--r-- | 6,731 | conv_transpose_op_mobile_test.cc |
|
- | rw-r--r-- | 10,250 | conv_transpose_unpool_op_base.h |
|
- | rw-r--r-- | 6,683 | copy_op.cc |
|
- | rw-r--r-- | 1,943 | copy_op.cu |
|
- | rw-r--r-- | 1,296 | copy_op.h |
|
- | rw-r--r-- | 2,446 | copy_rows_to_tensor_op.cc |
|
- | rw-r--r-- | 2,546 | copy_rows_to_tensor_op.h |
|
- | rw-r--r-- | 2,372 | cos_op.cc |
|
- | rw-r--r-- | 1,257 | cos_op.cu |
|
- | rw-r--r-- | 705 | cos_op.h |
|
- | rw-r--r-- | 2,418 | cosh_op.cc |
|
- | rw-r--r-- | 1,274 | cosh_op.cu |
|
- | rw-r--r-- | 711 | cosh_op.h |
|
- | rw-r--r-- | 2,570 | cosine_embedding_criterion_op.cc |
|
- | rw-r--r-- | 2,085 | cosine_embedding_criterion_op.cu |
|
- | rw-r--r-- | 1,127 | cosine_embedding_criterion_op.h |
|
- | rw-r--r-- | 8,969 | counter_ops.cc |
|
- | rw-r--r-- | 4,597 | counter_ops.h |
|
- | rw-r--r-- | 582 | counter_ops_gpu.cc |
|
- | rw-r--r-- | 560 | crash_op.cc |
|
- | rw-r--r-- | 1,241 | create_scope_op.cc |
|
- | rw-r--r-- | 5,233 | create_scope_op.h |
|
- | rw-r--r-- | 7,305 | crf_viterbi_op.cc |
|
- | rw-r--r-- | 22,632 | cross_entropy_op.cc |
|
- | rw-r--r-- | 13,733 | cross_entropy_op.cu |
|
- | rw-r--r-- | 4,422 | cross_entropy_op.h |
|
- | rw-r--r-- | 6,599 | ctc_beam_search_decoder_op.cc |
|
- | rw-r--r-- | 1,102 | ctc_beam_search_decoder_op.h |
|
- | rw-r--r-- | 3,030 | ctc_greedy_decoder_op.cc |
|
- | rw-r--r-- | 817 | ctc_greedy_decoder_op.h |
|
- | rw-r--r-- | 1,723 | cube_op.cc |
|
- | rw-r--r-- | 1,266 | cube_op.cu |
|
- | rw-r--r-- | 723 | cube_op.h |
|
- | rw-r--r-- | 590 | data_couple.cc |
|
- | rw-r--r-- | 464 | data_couple.h |
|
- | rw-r--r-- | 164 | data_couple_gpu.cu |
|
- | rw-r--r-- | 52,090 | dataset_ops.cc |
|
- | rw-r--r-- | 5,502 | dataset_ops.h |
|
- | rw-r--r-- | 1,725 | deform_conv_gradient_op.cc |
|
- | rw-r--r-- | 2,371 | deform_conv_op.cc |
|
- | rw-r--r-- | 21,554 | deform_conv_op.cu |
|
- | rw-r--r-- | 3,543 | deform_conv_op.h |
|
- | rw-r--r-- | 13,171 | deform_conv_op_impl.h |
|
- | rw-r--r-- | 1,182 | dense_vector_to_id_list_op.cc |
|
- | rw-r--r-- | 1,797 | dense_vector_to_id_list_op.h |
|
- | rw-r--r-- | 17,077 | depthwise_3x3_conv_op_cudnn.cu |
|
- | rw-r--r-- | 22,631 | distance_op.cc |
|
- | rw-r--r-- | 14,967 | distance_op.cu |
|
- | rw-r--r-- | 8,419 | distance_op.h |
|
- | rw-r--r-- | 1,193 | do_op.cc |
|
- | rw-r--r-- | 6,918 | do_op.h |
|
- | rw-r--r-- | 165 | do_op_gpu.cc |
|
- | rw-r--r-- | 5,597 | dropout_op.cc |
|
- | rw-r--r-- | 2,660 | dropout_op.cu |
|
- | rw-r--r-- | 1,516 | dropout_op.h |
|
- | rw-r--r-- | 8,576 | dropout_op_cudnn.cc |
|
- | rw-r--r-- | 683 | elementwise_add_gradient_op.cc |
|
- | rw-r--r-- | 201 | elementwise_add_op.cc |
|
- | rw-r--r-- | 1,792 | elementwise_add_op.h |
|
- | rw-r--r-- | 394 | elementwise_add_op_gpu.cc |
|
- | rw-r--r-- | 8,406 | elementwise_div_gradient_op.cc |
|
- | rw-r--r-- | 202 | elementwise_div_op.cc |
|
- | rw-r--r-- | 13,989 | elementwise_div_op.cu |
|
- | rw-r--r-- | 1,062 | elementwise_div_op.h |
|
- | rw-r--r-- | 5,254 | elementwise_linear_op.cc |
|
- | rw-r--r-- | 3,458 | elementwise_linear_op.cu |
|
- | rw-r--r-- | 1,170 | elementwise_linear_op.h |
|
- | rw-r--r-- | 3,042 | elementwise_logical_ops.cc |
|
- | rw-r--r-- | 5,083 | elementwise_logical_ops.h |
|
- | rw-r--r-- | 8,749 | elementwise_mul_gradient_op.cc |
|
- | rw-r--r-- | 201 | elementwise_mul_op.cc |
|
- | rw-r--r-- | 7,465 | elementwise_mul_op.cu |
|
- | rw-r--r-- | 1,062 | elementwise_mul_op.h |
|
- | rw-r--r-- | 975 | elementwise_op_gpu_test.cc |
|
- | rw-r--r-- | 764 | elementwise_op_test.cc |
|
- | rw-r--r-- | 9,237 | elementwise_op_test.h |
|
- | rw-r--r-- | 3,753 | elementwise_ops.cc |
|
- | rw-r--r-- | 7,306 | elementwise_ops.cu |
|
- | rw-r--r-- | 16,751 | elementwise_ops.h |
|
- | rw-r--r-- | 23,608 | elementwise_ops_schema.cc |
|
- | rw-r--r-- | 3,388 | elementwise_ops_utils.cc |
|
- | rw-r--r-- | 1,008 | elementwise_ops_utils.h |
|
- | rw-r--r-- | 683 | elementwise_sub_gradient_op.cc |
|
- | rw-r--r-- | 201 | elementwise_sub_op.cc |
|
- | rw-r--r-- | 1,793 | elementwise_sub_op.h |
|
- | rw-r--r-- | 394 | elementwise_sub_op_gpu.cc |
|
- | rw-r--r-- | 2,204 | elementwise_sum_op.cc |
|
- | rw-r--r-- | 3,805 | elu_op.cc |
|
- | rw-r--r-- | 2,152 | elu_op.cu |
|
- | rw-r--r-- | 875 | elu_op.h |
|
- | rw-r--r-- | 3,071 | elu_op_cudnn.cc |
|
- | rw-r--r-- | 524 | enforce_finite_op.cc |
|
- | rw-r--r-- | 374 | enforce_finite_op.cu |
|
- | rw-r--r-- | 1,141 | enforce_finite_op.h |
|
- | rw-r--r-- | 1,680 | ensure_clipped_op.cc |
|
- | rw-r--r-- | 1,608 | ensure_clipped_op.h |
|
- | rw-r--r-- | 1,139 | ensure_cpu_output_op.cc |
|
- | rw-r--r-- | 293 | ensure_cpu_output_op.cu |
|
- | rw-r--r-- | 1,465 | ensure_cpu_output_op.h |
|
- | rw-r--r-- | 1,803 | erf_op.cc |
|
- | rw-r--r-- | 1,326 | erf_op.cu |
|
- | rw-r--r-- | 751 | erf_op.h |
|
- | rw-r--r-- | 1,903 | exp_op.cc |
|
- | rw-r--r-- | 425 | exp_op.h |
|
- | rw-r--r-- | 260 | exp_op_gpu.cc |
|
- | rw-r--r-- | 1,724 | expand_op.cc |
|
- | rw-r--r-- | 3,543 | expand_op.h |
|
- | rw-r--r-- | 411 | expand_op_gpu.cc |
|
- | rw-r--r-- | 5,867 | expand_squeeze_dims_op.cc |
|
- | rw-r--r-- | 3,389 | expand_squeeze_dims_op.h |
|
- | rw-r--r-- | 252 | expand_squeeze_dims_op_gpu.cc |
|
- | rw-r--r-- | 4,567 | fc_inference.cc |
|
- | rw-r--r-- | 775 | fc_inference.h |
|
- | rw-r--r-- | 14,801 | feature_maps_ops.cc |
|
- | rw-r--r-- | 32,437 | feature_maps_ops.h |
|
- | rw-r--r-- | 508 | feed_blob_op.cc |
|
- | rw-r--r-- | 802 | feed_blob_op.h |
|
- | rw-r--r-- | 21,486 | filler_op.cc |
|
- | rw-r--r-- | 2,156 | filler_op.cu |
|
- | rw-r--r-- | 18,409 | filler_op.h |
|
- | rw-r--r-- | 1,509 | find_duplicate_elements_op.cc |
|
- | rw-r--r-- | 1,563 | find_duplicate_elements_op.h |
|
- | rw-r--r-- | 739 | find_op.cc |
|
- | rw-r--r-- | 1,483 | find_op.cu |
|
- | rw-r--r-- | 2,055 | find_op.h |
|
- | rw-r--r-- | 2,178 | flatten_op.cc |
|
- | rw-r--r-- | 1,525 | flatten_op.h |
|
- | rw-r--r-- | 5,785 | flexible_top_k.cc |
|
- | rw-r--r-- | 936 | flexible_top_k.h |
|
- | rw-r--r-- | 1,719 | floor_op.cc |
|
- | rw-r--r-- | 745 | floor_op.cu |
|
- | rw-r--r-- | 788 | floor_op.h |
|
- | rw-r--r-- | 408 | free_op.cc |
|
- | rw-r--r-- | 777 | free_op.h |
|
- | rw-r--r-- | 168 | free_op_gpu.cc |
|
- | rw-r--r-- | 6,096 | fully_connected_op.cc |
|
- | rw-r--r-- | 9,351 | fully_connected_op.h |
|
- | rw-r--r-- | 6,059 | fully_connected_op_gpu.cc |
|
- | rw-r--r-- | 12,811 | fused_rowwise_8bit_conversion_ops.cc |
|
- | rw-r--r-- | 6,491 | fused_rowwise_8bit_conversion_ops.h |
|
- | rw-r--r-- | 11,672 | fused_rowwise_nbit_conversion_ops.cc |
|
- | rw-r--r-- | 8,668 | fused_rowwise_nbit_conversion_ops.h |
|
- | rw-r--r-- | 8,065 | fused_rowwise_nbitfake_conversion_ops.cc |
|
- | rw-r--r-- | 4,320 | fused_rowwise_nbitfake_conversion_ops.h |
|
- | rw-r--r-- | 8,316 | fused_rowwise_random_quantization_ops.cc |
|
- | rw-r--r-- | 2,607 | fused_rowwise_random_quantization_ops.h |
|
- | rw-r--r-- | 1,272 | gather_fused_8bit_rowwise_op.cc |
|
- | rw-r--r-- | 2,179 | gather_fused_8bit_rowwise_op.h |
|
- | rw-r--r-- | 4,282 | gather_op.cc |
|
- | rw-r--r-- | 646 | gather_op.cu |
|
- | rw-r--r-- | 3,861 | gather_op.cuh |
|
- | rw-r--r-- | 7,505 | gather_op.h |
|
- | rw-r--r-- | 3,848 | gather_ranges_to_dense_op.cc |
|
- | rw-r--r-- | 7,966 | gather_ranges_to_dense_op.h |
|
- | rw-r--r-- | 3,720 | gelu_op.cc |
|
- | rw-r--r-- | 6,793 | gelu_op.cu |
|
- | rw-r--r-- | 1,452 | gelu_op.h |
|
- | rw-r--r-- | 16,584 | generate_proposals_op.cc |
|
- | rw-r--r-- | 24,075 | generate_proposals_op.cu |
|
- | rw-r--r-- | 6,258 | generate_proposals_op.h |
|
- | rw-r--r-- | 29,976 | generate_proposals_op_gpu_test.cc |
|
- | rw-r--r-- | 33,161 | generate_proposals_op_test.cc |
|
- | rw-r--r-- | 14,271 | generate_proposals_op_util_boxes.h |
|
- | rw-r--r-- | 5,378 | generate_proposals_op_util_boxes_test.cc |
|
- | rw-r--r-- | 25,934 | generate_proposals_op_util_nms.h |
|
- | rw-r--r-- | 18,901 | generate_proposals_op_util_nms_gpu.cu |
|
- | rw-r--r-- | 2,131 | generate_proposals_op_util_nms_gpu.h |
|
- | rw-r--r-- | 22,491 | generate_proposals_op_util_nms_gpu_test.cc |
|
- | rw-r--r-- | 18,893 | generate_proposals_op_util_nms_test.cc |
|
- | rw-r--r-- | 2,299 | given_tensor_byte_string_to_uint8_fill_op.cc |
|
- | rw-r--r-- | 248 | given_tensor_byte_string_to_uint8_fill_op.cu |
|
- | rw-r--r-- | 2,097 | given_tensor_byte_string_to_uint8_fill_op.h |
|
- | rw-r--r-- | 8,332 | given_tensor_fill_op.cc |
|
- | rw-r--r-- | 652 | given_tensor_fill_op.cu |
|
- | rw-r--r-- | 2,949 | given_tensor_fill_op.h |
|
- | rw-r--r-- | 1,474 | glu_op.cc |
|
- | rw-r--r-- | 1,142 | glu_op.cu |
|
- | rw-r--r-- | 1,458 | glu_op.h |
|
- | rw-r--r-- | 14,500 | group_norm_op.cc |
|
- | rw-r--r-- | 16,508 | group_norm_op.cu |
|
- | rw-r--r-- | 8,967 | group_norm_op.h |
|
- | rw-r--r-- | 1,894 | gru_unit_op.cc |
|
- | rw-r--r-- | 6,626 | gru_unit_op.h |
|
- | rw-r--r-- | 3,572 | gru_unit_op_gpu.cu |
|
- | rw-r--r-- | 23,221 | h_softmax_op.cc |
|
- | rw-r--r-- | 4,954 | h_softmax_op.h |
|
- | rw-r--r-- | 5,675 | half_float_ops.cc |
|
- | rw-r--r-- | 2,418 | half_float_ops.cu |
|
- | rw-r--r-- | 2,732 | half_float_ops.h |
|
- | rw-r--r-- | 3,102 | half_float_ops_test.cc |
|
- | rw-r--r-- | 3,852 | hard_sigmoid_op.cc |
|
- | rw-r--r-- | 2,181 | hard_sigmoid_op.cu |
|
- | rw-r--r-- | 994 | hard_sigmoid_op.h |
|
- | rw-r--r-- | 6,692 | heatmap_max_keypoint_op.cc |
|
- | rw-r--r-- | 939 | heatmap_max_keypoint_op.h |
|
- | rw-r--r-- | 1,229 | histogram_op.cc |
|
- | rw-r--r-- | 2,311 | histogram_op.h |
|
- | rw-r--r-- | 734 | if_op.cc |
|
- | rw-r--r-- | 1,764 | if_op.h |
|
- | rw-r--r-- | 165 | if_op_gpu.cc |
|
- | rw-r--r-- | 3,835 | im2col_op.cc |
|
- | rw-r--r-- | 8,943 | im2col_op.h |
|
- | rw-r--r-- | 245 | im2col_op_gpu.cc |
|
- | rw-r--r-- | 1,364 | index_hash_ops.cc |
|
- | rw-r--r-- | 2,179 | index_hash_ops.h |
|
- | rw-r--r-- | 11,534 | index_ops.cc |
|
- | rw-r--r-- | 3,047 | index_ops.h |
|
- | rw-r--r-- | 1,918 | inference_lstm_op.cc |
|
- | rw-r--r-- | 9,881 | inference_lstm_op.h |
|
- | rw-r--r-- | 7,816 | instance_norm_gradient_op.cc |
|
- | rw-r--r-- | 6,485 | instance_norm_op.cc |
|
- | rw-r--r-- | 13,428 | instance_norm_op.cu |
|
- | rw-r--r-- | 7,441 | instance_norm_op.h |
|
- | rw-r--r-- | 5,031 | integral_image_op.cc |
|
- | rw-r--r-- | 6,808 | integral_image_op.cu |
|
- | rw-r--r-- | 923 | integral_image_op.h |
|
- | rw-r--r-- | 1,557 | is_empty_op.cc |
|
- | rw-r--r-- | 558 | is_empty_op.h |
|
- | rw-r--r-- | 2,848 | jsd_op.cc |
|
- | rw-r--r-- | 721 | jsd_op.h |
|
- | rw-r--r-- | 308 | key_split_ops.cc |
|
- | rw-r--r-- | 1,400 | key_split_ops.h |
|
- | rw-r--r-- | 6,033 | last_n_window_collector.cc |
|
- | rw-r--r-- | 9,381 | layer_norm_op.cc |
|
- | rw-r--r-- | 10,483 | layer_norm_op.cu |
|
- | rw-r--r-- | 8,042 | layer_norm_op.h |
|
- | rw-r--r-- | 3,440 | leaky_relu_op.cc |
|
- | rw-r--r-- | 1,667 | leaky_relu_op.cu |
|
- | rw-r--r-- | 1,111 | leaky_relu_op.h |
|
- | rw-r--r-- | 1,194 | length_split_op.cc |
|
- | rw-r--r-- | 2,259 | length_split_op.h |
|
- | rw-r--r-- | 1,861 | lengths_pad_op.cc |
|
- | rw-r--r-- | 188 | lengths_pad_op.cu |
|
- | rw-r--r-- | 2,574 | lengths_pad_op.h |
|
- | rw-r--r-- | 3,974 | lengths_reducer_fused_8bit_rowwise_ops.cc |
|
- | rw-r--r-- | 5,532 | lengths_reducer_fused_8bit_rowwise_ops.h |
|
- | rw-r--r-- | 21,990 | lengths_reducer_fused_nbit_rowwise_ops.cc |
|
- | rw-r--r-- | 23,433 | lengths_reducer_fused_nbit_rowwise_ops.h |
|
- | rw-r--r-- | 7,618 | lengths_reducer_ops.cc |
|
- | rw-r--r-- | 23,259 | lengths_reducer_ops.h |
|
- | rw-r--r-- | 8,053 | lengths_reducer_rowwise_8bit_ops.cc |
|
- | rw-r--r-- | 6,180 | lengths_reducer_rowwise_8bit_ops.h |
|
- | rw-r--r-- | 2,821 | lengths_tile_op.cc |
|
- | rw-r--r-- | 3,299 | lengths_tile_op.cu |
|
- | rw-r--r-- | 582 | lengths_tile_op.h |
|
- | rw-r--r-- | 5,233 | lengths_top_k_op.cc |
|
- | rw-r--r-- | 1,358 | lengths_top_k_op.h |
|
- | rw-r--r-- | 8,836 | listwise_l2r_op.cc |
|
- | rw-r--r-- | 1,677 | listwise_l2r_op.h |
|
- | rw-r--r-- | 9,943 | load_save_op.cc |
|
- | rw-r--r-- | 15,847 | load_save_op.h |
|
- | rw-r--r-- | 632 | load_save_op_gpu.cc |
|
- | rw-r--r-- | 4,645 | load_save_op_util.cc |
|
- | rw-r--r-- | 1,647 | load_save_op_util.h |
|
- | rw-r--r-- | 15,328 | local_response_normalization_op.cc |
|
- | rw-r--r-- | 11,528 | local_response_normalization_op.cu |
|
- | rw-r--r-- | 2,804 | local_response_normalization_op.h |
|
- | rw-r--r-- | 5,804 | local_response_normalization_op_cudnn.cc |
|
- | rw-r--r-- | 5,364 | locally_connected_op.cc |
|
- | rw-r--r-- | 3,872 | locally_connected_op.h |
|
- | rw-r--r-- | 847 | locally_connected_op_gpu.cc |
|
- | rw-r--r-- | 26,495 | locally_connected_op_impl.h |
|
- | rw-r--r-- | 2,108 | locally_connected_op_util.cc |
|
- | rw-r--r-- | 1,334 | locally_connected_op_util.h |
|
- | rw-r--r-- | 1,936 | log_op.cc |
|
- | rw-r--r-- | 431 | log_op.h |
|
- | rw-r--r-- | 260 | log_op_gpu.cc |
|
- | rw-r--r-- | 2,957 | logit_op.cc |
|
- | rw-r--r-- | 1,668 | logit_op.cu |
|
- | rw-r--r-- | 1,129 | logit_op.h |
|
- | rw-r--r-- | 1,844 | loss_op.cc |
|
- | rw-r--r-- | 287 | loss_op.cu |
|
- | rw-r--r-- | 1,058 | loss_op.h |
|
- | rw-r--r-- | 10,547 | lp_pool_op.cc |
|
- | rw-r--r-- | 10,498 | lp_pool_op.cu |
|
- | rw-r--r-- | 5,151 | lpnorm_op.cc |
|
- | rw-r--r-- | 275 | lpnorm_op.cu |
|
- | rw-r--r-- | 1,279 | lpnorm_op.h |
|
- | rw-r--r-- | 1,811 | lstm_unit_op.cc |
|
- | rw-r--r-- | 6,733 | lstm_unit_op.h |
|
- | rw-r--r-- | 7,147 | lstm_unit_op_gpu.cu |
|
- | rw-r--r-- | 9,424 | lstm_utils.h |
|
- | rw-r--r-- | 2,494 | map_ops.cc |
|
- | rw-r--r-- | 8,011 | map_ops.h |
|
- | rw-r--r-- | 3,521 | margin_ranking_criterion_op.cc |
|
- | rw-r--r-- | 2,633 | margin_ranking_criterion_op.cu |
|
- | rw-r--r-- | 1,113 | margin_ranking_criterion_op.h |
|
- | rw-r--r-- | 7,097 | matmul_op.cc |
|
- | rw-r--r-- | 2,843 | matmul_op.h |
|
- | rw-r--r-- | 164 | matmul_op_gpu.cc |
|
- | rw-r--r-- | 7,756 | max_pool_with_index.cu |
|
- | rw-r--r-- | 1,155 | max_pool_with_index_gpu.h |
|
- | rw-r--r-- | 2,548 | mean_op.cc |
|
- | rw-r--r-- | 237 | mean_op.cu |
|
- | rw-r--r-- | 3,252 | mean_op.h |
|
- | rw-r--r-- | 1,548 | mem_query_op.cu |
|
- | rw-r--r-- | 1,637 | merge_id_lists_op.cc |
|
- | rw-r--r-- | 2,517 | merge_id_lists_op.h |
|
- | rw-r--r-- | 2,107 | minmax_gradient_ops.cc |
|
- | rw-r--r-- | 3,717 | minmax_ops.cc |
|
- | rw-r--r-- | 1,586 | minmax_ops.cu |
|
- | rw-r--r-- | 3,829 | minmax_ops.h |
|
- | rw-r--r-- | 2,903 | mish_op.cc |
|
- | rw-r--r-- | 766 | mish_op.h |
|
- | rw-r--r-- | 2,276 | mod_op.cc |
|
- | rw-r--r-- | 984 | mod_op.h |
|
- | rw-r--r-- | 2,600 | moments_op.cc |
|
- | rw-r--r-- | 3,080 | moments_op.cu |
|
- | rw-r--r-- | 3,756 | moments_op.h |
|
- | rw-r--r-- | 2,295 | multi_class_accuracy_op.cc |
|
- | rw-r--r-- | 2,175 | multi_class_accuracy_op.cu |
|
- | rw-r--r-- | 539 | multi_class_accuracy_op.h |
|
- | rw-r--r-- | 802 | negate_gradient_op.cc |
|
- | rw-r--r-- | 566 | negate_gradient_op.h |
|
- | rw-r--r-- | 198 | negate_gradient_op_gpu.cc |
|
- | rw-r--r-- | 1,709 | negative_op.cc |
|
- | rw-r--r-- | 451 | negative_op.h |
|
- | rw-r--r-- | 269 | negative_op_gpu.cc |
|
- | rw-r--r-- | 357 | ngram_ops.cc |
|
- | rw-r--r-- | 2,644 | ngram_ops.h |
|
- | rw-r--r-- | 1,063 | no_default_engine_op.h |
|
- | rw-r--r-- | 1,575 | norm_planar_yuv_op.cc |
|
- | rw-r--r-- | 1,156 | normalize_l1_op.cc |
|
- | rw-r--r-- | 1,075 | normalize_l1_op.h |
|
- | rw-r--r-- | 2,010 | normalize_op.cc |
|
- | rw-r--r-- | 3,013 | normalize_op.h |
|
- | rw-r--r-- | 5,009 | normalize_ops.cu |
|
- | rw-r--r-- | 503 | numpy_tile_op.cc |
|
- | rw-r--r-- | 3,591 | numpy_tile_op.h |
|
- | rw-r--r-- | 12,121 | one_hot_ops.cc |
|
- | rw-r--r-- | 929 | one_hot_ops.cu |
|
- | rw-r--r-- | 2,562 | one_hot_ops.h |
|
- | rw-r--r-- | 3,351 | onnx_while_op.cc |
|
- | rw-r--r-- | 10,600 | onnx_while_op.h |
|
- | rw-r--r-- | 2,112 | op_utils_cudnn.h |
|
- | rw-r--r-- | 4,155 | operator_fallback_gpu.h |
|
- | rw-r--r-- | 2,580 | operator_fallback_gpu_test.cc |
|
- | rw-r--r-- | 3,120 | order_switch_ops.cc |
|
- | rw-r--r-- | 2,149 | order_switch_ops.h |
|
- | rw-r--r-- | 5,871 | order_switch_ops_cudnn.cc |
|
- | rw-r--r-- | 265 | order_switch_ops_gpu.cc |
|
- | rw-r--r-- | 3,476 | pack_rnn_sequence_op.cc |
|
- | rw-r--r-- | 3,074 | pack_rnn_sequence_op.h |
|
- | rw-r--r-- | 8,499 | pack_segments.cc |
|
- | rw-r--r-- | 9,818 | pack_segments.cu |
|
- | rw-r--r-- | 2,729 | pack_segments.h |
|
- | rw-r--r-- | 15,401 | pad_op.cc |
|
- | rw-r--r-- | 2,903 | pad_op.h |
|
- | rw-r--r-- | 16,638 | pad_op_gpu.cu |
|
- | rw-r--r-- | 5,455 | partition_ops.cc |
|
- | rw-r--r-- | 9,958 | partition_ops.h |
|
- | rw-r--r-- | 6,162 | percentile_op.cc |
|
- | rw-r--r-- | 1,009 | percentile_op.h |
|
- | rw-r--r-- | 1,259 | perplexity_op.cc |
|
- | rw-r--r-- | 1,190 | perplexity_op.cu |
|
- | rw-r--r-- | 447 | perplexity_op.h |
|
- | rw-r--r-- | 3,945 | piecewise_linear_transform_op.cc |
|
- | rw-r--r-- | 8,243 | piecewise_linear_transform_op.cu |
|
- | rw-r--r-- | 8,281 | piecewise_linear_transform_op.h |
|
- | rw-r--r-- | 26,669 | pool_gradient_op.cc |
|
- | rw-r--r-- | 42,560 | pool_op.cc |
|
- | rw-r--r-- | 60,286 | pool_op.cu |
|
- | rw-r--r-- | 8,525 | pool_op.h |
|
- | rw-r--r-- | 16,580 | pool_op_cudnn.cc |
|
- | rw-r--r-- | 11,287 | pool_op_util.cc |
|
- | rw-r--r-- | 1,105 | pool_op_util.h |
|
- | rw-r--r-- | 13,012 | pow_op.cc |
|
- | rw-r--r-- | 2,660 | pow_op.cu |
|
- | rw-r--r-- | 4,677 | pow_op.h |
|
- | rw-r--r-- | 4,661 | prefetch_op.h |
|
- | rw-r--r-- | 10,346 | prelu_op.cc |
|
- | rw-r--r-- | 7,428 | prelu_op.cu |
|
- | rw-r--r-- | 1,067 | prelu_op.h |
|
- | rw-r--r-- | 1,342 | prepend_dim_op.cc |
|
- | rw-r--r-- | 2,654 | prepend_dim_op.h |
|
- | rw-r--r-- | 248 | prepend_dim_op_gpu.cc |
|
- | rw-r--r-- | 2,147 | quant_decode_op.cc |
|
- | rw-r--r-- | 5,370 | quant_decode_op.h |
|
- | rw-r--r-- | 614 | quantile_op.cc |
|
- | rw-r--r-- | 4,120 | quantile_op.h |
|
- | rw-r--r-- | 6,217 | rank_loss_op.cc |
|
- | rw-r--r-- | 820 | rank_loss_op.h |
|
- | rw-r--r-- | 1,311 | reciprocal_gradient_op.cc |
|
- | rw-r--r-- | 1,345 | reciprocal_op.cc |
|
- | rw-r--r-- | 1,341 | reciprocal_op.cu |
|
- | rw-r--r-- | 721 | reciprocal_op.h |
|
- | rw-r--r-- | 8,403 | reduce_front_back_max_ops.cc |
|
- | rw-r--r-- | 4,769 | reduce_front_back_max_ops.cu |
|
- | rw-r--r-- | 4,399 | reduce_front_back_max_ops.h |
|
- | rw-r--r-- | 8,600 | reduce_front_back_mean_ops.cc |
|
- | rw-r--r-- | 7,047 | reduce_front_back_sum_mean_ops.cu |
|
- | rw-r--r-- | 5,337 | reduce_front_back_sum_mean_ops.h |
|
- | rw-r--r-- | 8,300 | reduce_front_back_sum_ops.cc |
|
- | rw-r--r-- | 15,216 | reduce_ops.cc |
|
- | rw-r--r-- | 4,508 | reduce_ops.cu |
|
- | rw-r--r-- | 8,565 | reduce_ops.h |
|
- | rw-r--r-- | 24,556 | reducer_functors.h |
|
- | rw-r--r-- | 9,524 | reduction_ops.cc |
|
- | rw-r--r-- | 3,736 | reduction_ops.cu |
|
- | rw-r--r-- | 5,944 | reduction_ops.h |
|
- | rw-r--r-- | 2,785 | relu_n_op.cc |
|
- | rw-r--r-- | 2,057 | relu_n_op.cu |
|
- | rw-r--r-- | 990 | relu_n_op.h |
|
- | rw-r--r-- | 3,995 | relu_op.cc |
|
- | rw-r--r-- | 7,038 | relu_op.cu |
|
- | rw-r--r-- | 624 | relu_op.h |
|
- | rw-r--r-- | 798 | remove_data_blocks_op.cc |
|
- | rw-r--r-- | 2,651 | remove_data_blocks_op.h |
|
- | rw-r--r-- | 854 | replace_nan_op.cc |
|
- | rw-r--r-- | 745 | replace_nan_op.cu |
|
- | rw-r--r-- | 1,170 | replace_nan_op.h |
|
- | rw-r--r-- | 9,048 | reservoir_sampling.cc |
|
- | rw-r--r-- | 5,721 | reshape_op.cc |
|
- | rw-r--r-- | 5,668 | reshape_op.h |
|
- | rw-r--r-- | 217 | reshape_op_gpu.cc |
|
- | rw-r--r-- | 1,457 | reshape_op_gpu_test.cc |
|
- | rw-r--r-- | 7,081 | resize_3d_op.cc |
|
- | rw-r--r-- | 6,372 | resize_3d_op.cu |
|
- | rw-r--r-- | 2,677 | resize_3d_op.h |
|
- | rw-r--r-- | 11,405 | resize_op.cc |
|
- | rw-r--r-- | 4,961 | resize_op.cu |
|
- | rw-r--r-- | 2,307 | resize_op.h |
|
- | rw-r--r-- | 1,119 | reverse_packed_segs_op.cc |
|
- | rw-r--r-- | 2,530 | reverse_packed_segs_op.cu |
|
- | rw-r--r-- | 2,772 | reverse_packed_segs_op.h |
|
- | rw-r--r-- | 3,643 | rmac_regions_op.cc |
|
- | rw-r--r-- | 5,987 | rmac_regions_op.cu |
|
- | rw-r--r-- | 708 | rmac_regions_op.h |
|
- | rw-r--r-- | 5,514 | rms_norm_op.cc |
|
- | rw-r--r-- | 5,173 | rms_norm_op.cu |
|
- | rw-r--r-- | 2,968 | rms_norm_op.h |
|
- | rw-r--r-- | 8,220 | roi_align_gradient_op.cc |
|
- | rw-r--r-- | 7,226 | roi_align_gradient_op.cu |
|
- | rw-r--r-- | 1,486 | roi_align_gradient_op.h |
|
- | rw-r--r-- | 11,075 | roi_align_op.cc |
|
- | rw-r--r-- | 5,548 | roi_align_op.cu |
|
- | rw-r--r-- | 2,857 | roi_align_op.h |
|
- | rw-r--r-- | 9,005 | roi_align_op_gpu_test.cc |
|
- | rw-r--r-- | 924 | roi_align_rotated_gradient_op.cc |
|
- | rw-r--r-- | 7,176 | roi_align_rotated_gradient_op.cu |
|
- | rw-r--r-- | 1,369 | roi_align_rotated_gradient_op.h |
|
- | rw-r--r-- | 13,481 | roi_align_rotated_op.cc |
|
- | rw-r--r-- | 6,105 | roi_align_rotated_op.cu |
|
- | rw-r--r-- | 1,636 | roi_align_rotated_op.h |
|
- | rw-r--r-- | 7,358 | roi_pool_op.cc |
|
- | rw-r--r-- | 6,904 | roi_pool_op.cu |
|
- | rw-r--r-- | 2,503 | roi_pool_op.h |
|
- | rw-r--r-- | 1,550 | rowmul_op.cc |
|
- | rw-r--r-- | 1,947 | rowmul_op.h |
|
- | rw-r--r-- | 1,758 | rsqrt_op.cc |
|
- | rw-r--r-- | 1,410 | rsqrt_op.cu |
|
- | rw-r--r-- | 729 | rsqrt_op.h |
|
- | rw-r--r-- | 553 | scale_blobs_op.cc |
|
- | rw-r--r-- | 5,274 | scale_blobs_op.cu |
|
- | rw-r--r-- | 1,458 | scale_blobs_op.h |
|
- | rw-r--r-- | 852 | scale_op.cc |
|
- | rw-r--r-- | 1,019 | scale_op.h |
|
- | rw-r--r-- | 309 | scale_op_gpu.cc |
|
- | rw-r--r-- | 21,543 | segment_reduction_op.cc |
|
- | rw-r--r-- | 70,726 | segment_reduction_op.h |
|
- | rw-r--r-- | 57,761 | segment_reduction_op_gpu.cu |
|
- | rw-r--r-- | 2,801 | segment_reduction_op_gpu.cuh |
|
- | rw-r--r-- | 1,748 | self_binning_histogram_op.cc |
|
- | rw-r--r-- | 6,011 | self_binning_histogram_op.h |
|
- | rw-r--r-- | 4,131 | selu_op.cc |
|
- | rw-r--r-- | 1,884 | selu_op.cu |
|
- | rw-r--r-- | 1,545 | selu_op.h |
|
- | rw-r--r-- | 16,078 | sequence_ops.cc |
|
- | rw-r--r-- | 10,875 | sequence_ops.cu |
|
- | rw-r--r-- | 8,264 | sequence_ops.h |
|
- | rw-r--r-- | 1,769 | shape_op.cc |
|
- | rw-r--r-- | 1,638 | shape_op.h |
|
- | rw-r--r-- | 151 | shape_op_gpu.cc |
|
- | rw-r--r-- | 1,297 | sigmoid_gradient_op.cc |
|
- | rw-r--r-- | 1,997 | sigmoid_op.cc |
|
- | rw-r--r-- | 1,977 | sigmoid_op.cu |
|
- | rw-r--r-- | 639 | sigmoid_op.h |
|
- | rw-r--r-- | 320 | sigmoid_op_cudnn.cc |
|
- | rw-r--r-- | 2,346 | sin_op.cc |
|
- | rw-r--r-- | 1,267 | sin_op.cu |
|
- | rw-r--r-- | 705 | sin_op.h |
|
- | rw-r--r-- | 2,414 | sinh_op.cc |
|
- | rw-r--r-- | 1,274 | sinh_op.cu |
|
- | rw-r--r-- | 711 | sinh_op.h |
|
- | rw-r--r-- | 1,404 | sinusoid_position_encoding_op.cc |
|
- | rw-r--r-- | 2,834 | sinusoid_position_encoding_op.h |
|
- | rw-r--r-- | 3,945 | slice_op.cc |
|
- | rw-r--r-- | 10,846 | slice_op.cu |
|
- | rw-r--r-- | 10,069 | slice_op.h |
|
- | rw-r--r-- | 5,772 | softmax_op.cc |
|
- | rw-r--r-- | 1,174 | softmax_op.h |
|
- | rw-r--r-- | 4,048 | softmax_op_cudnn.cc |
|
- | rw-r--r-- | 23,833 | softmax_ops.cu |
|
- | rw-r--r-- | 1,867 | softmax_utils.cc |
|
- | rw-r--r-- | 447 | softmax_utils.h |
|
- | rw-r--r-- | 12,661 | softmax_with_loss_op.cc |
|
- | rw-r--r-- | 2,883 | softmax_with_loss_op.h |
|
- | rw-r--r-- | 2,977 | softplus_op.cc |
|
- | rw-r--r-- | 1,609 | softplus_op.cu |
|
- | rw-r--r-- | 781 | softplus_op.h |
|
- | rw-r--r-- | 3,623 | softsign_op.cc |
|
- | rw-r--r-- | 2,031 | softsign_op.cu |
|
- | rw-r--r-- | 675 | softsign_op.h |
|
- | rw-r--r-- | 3,909 | space_batch_op.cc |
|
- | rw-r--r-- | 6,848 | space_batch_op.h |
|
- | rw-r--r-- | 4,852 | space_batch_op_gpu.cu |
|
- | rw-r--r-- | 3,922 | sparse_dropout_with_replacement_op.cc |
|
- | rw-r--r-- | 1,122 | sparse_dropout_with_replacement_op.h |
|
- | rw-r--r-- | 3,537 | sparse_lp_regularizer_op.cc |
|
- | rw-r--r-- | 1,131 | sparse_lp_regularizer_op.h |
|
- | rw-r--r-- | 226 | sparse_lp_regularizer_op_gpu.cu |
|
- | rw-r--r-- | 2,524 | sparse_normalize_op.cc |
|
- | rw-r--r-- | 835 | sparse_normalize_op.h |
|
- | rw-r--r-- | 226 | sparse_normalize_op_gpu.cu |
|
- | rw-r--r-- | 4,261 | sparse_to_dense_mask_op.cc |
|
- | rw-r--r-- | 9,784 | sparse_to_dense_mask_op.h |
|
- | rw-r--r-- | 2,394 | sparse_to_dense_op.cc |
|
- | rw-r--r-- | 2,387 | sparse_to_dense_op.cu |
|
- | rw-r--r-- | 3,977 | sparse_to_dense_op.h |
|
- | rw-r--r-- | 6,326 | spatial_batch_norm_gradient_op.cc |
|
- | rw-r--r-- | 6,003 | spatial_batch_norm_op.cc |
|
- | rw-r--r-- | 294 | spatial_batch_norm_op.cu |
|
- | rw-r--r-- | 15,175 | spatial_batch_norm_op.h |
|
- | rw-r--r-- | 12,449 | spatial_batch_norm_op_cudnn.cu |
|
- | rw-r--r-- | 12,159 | spatial_batch_norm_op_impl.cuh |
|
- | rw-r--r-- | 8,362 | spatial_softmax_with_loss_op.cc |
|
- | rw-r--r-- | 2,182 | spatial_softmax_with_loss_op.h |
|
- | rw-r--r-- | 2,107 | sqr_op.cc |
|
- | rw-r--r-- | 431 | sqr_op.h |
|
- | rw-r--r-- | 260 | sqr_op_gpu.cc |
|
- | rw-r--r-- | 2,234 | sqrt_op.cc |
|
- | rw-r--r-- | 448 | sqrt_op.h |
|
- | rw-r--r-- | 263 | sqrt_op_gpu.cc |
|
- | rw-r--r-- | 1,083 | square_root_divide_op.cc |
|
- | rw-r--r-- | 1,804 | square_root_divide_op.h |
|
- | rw-r--r-- | 9,945 | stats_ops.cc |
|
- | rw-r--r-- | 3,630 | stats_put_ops.cc |
|
- | rw-r--r-- | 2,757 | stats_put_ops.h |
|
- | rw-r--r-- | 551 | stop_gradient.cc |
|
- | rw-r--r-- | 548 | stop_gradient.h |
|
- | rw-r--r-- | 191 | stop_gradient_gpu.cc |
|
- | rw-r--r-- | 6,210 | string_ops.cc |
|
- | rw-r--r-- | 2,067 | string_ops.h |
|
- | rw-r--r-- | 4,072 | string_ops_test.cc |
|
- | rw-r--r-- | 3,038 | stump_func_op.cc |
|
- | rw-r--r-- | 1,589 | stump_func_op.cu |
|
- | rw-r--r-- | 2,112 | stump_func_op.h |
|
- | rw-r--r-- | 19,487 | stylizer_ops.cc |
|
- | rw-r--r-- | 2,285 | summarize_op.cc |
|
- | rw-r--r-- | 3,386 | summarize_op.cu |
|
- | rw-r--r-- | 1,875 | summarize_op.h |
|
- | rw-r--r-- | 2,662 | swish_op.cc |
|
- | rw-r--r-- | 2,128 | swish_op.cu |
|
- | rw-r--r-- | 772 | swish_op.h |
|
- | rw-r--r-- | 1,742 | tan_op.cc |
|
- | rw-r--r-- | 1,345 | tan_op.cu |
|
- | rw-r--r-- | 705 | tan_op.h |
|
- | rw-r--r-- | 1,286 | tanh_gradient_op.cc |
|
- | rw-r--r-- | 1,900 | tanh_op.cc |
|
- | rw-r--r-- | 1,305 | tanh_op.cu |
|
- | rw-r--r-- | 723 | tanh_op.h |
|
- | rw-r--r-- | 305 | tanh_op_cudnn.cc |
|
- | rw-r--r-- | 1,701 | tensor_protos_db_input.cc |
|
- | rw-r--r-- | 3,633 | tensor_protos_db_input.h |
|
- | rw-r--r-- | 248 | tensor_protos_db_input_gpu.cc |
|
- | rw-r--r-- | 6,421 | text_file_reader.cc |
|
- | rw-r--r-- | 3,121 | text_file_reader_utils.cc |
|
- | rw-r--r-- | 2,907 | text_file_reader_utils.h |
|
- | rw-r--r-- | 4,261 | text_file_reader_utils_test.cc |
|
- | rw-r--r-- | 2,902 | thresholded_relu_op.cc |
|
- | rw-r--r-- | 1,642 | thresholded_relu_op.cu |
|
- | rw-r--r-- | 1,137 | thresholded_relu_op.h |
|
- | rw-r--r-- | 6,111 | tile_op.cc |
|
- | rw-r--r-- | 2,723 | tile_op.cu |
|
- | rw-r--r-- | 8,741 | tile_op.h |
|
- | rw-r--r-- | 10,167 | top_k.cc |
|
- | rw-r--r-- | 10,794 | top_k.cu |
|
- | rw-r--r-- | 1,061 | top_k.h |
|
- | rw-r--r-- | 8,949 | top_k_heap_selection.cuh |
|
- | rw-r--r-- | 15,557 | top_k_radix_selection.cuh |
|
- | rw-r--r-- | 3,434 | transpose_op.cc |
|
- | rw-r--r-- | 186 | transpose_op.cu |
|
- | rw-r--r-- | 2,043 | transpose_op.h |
|
- | rw-r--r-- | 5,143 | transpose_op_cudnn.cc |
|
- | rw-r--r-- | 2,569 | tt_linear_op.cc |
|
- | rw-r--r-- | 6,448 | tt_linear_op.h |
|
- | rw-r--r-- | 3,534 | unique_ops.cc |
|
- | rw-r--r-- | 4,280 | unique_ops.cu |
|
- | rw-r--r-- | 1,666 | unique_ops.h |
|
- | rw-r--r-- | 6,978 | upsample_op.cc |
|
- | rw-r--r-- | 7,546 | upsample_op.cu |
|
- | rw-r--r-- | 2,246 | upsample_op.h |
|
- | rw-r--r-- | 33,252 | utility_ops.cc |
|
- | rw-r--r-- | 10,469 | utility_ops.cu |
|
- | rw-r--r-- | 48,170 | utility_ops.h |
|
- | rw-r--r-- | 1,427 | utility_ops_gpu_test.cc |
|
- | rw-r--r-- | 1,223 | utility_ops_test.cc |
|
- | rw-r--r-- | 713 | variable_length_sequence_padding.cc |
|
- | rw-r--r-- | 1,378 | variable_length_sequence_padding.h |
|
- | rw-r--r-- | 4,034 | weighted_multi_sampling_op.cc |
|
- | rw-r--r-- | 602 | weighted_multi_sampling_op.h |
|
- | rw-r--r-- | 4,109 | weighted_sample_op.cc |
|
- | rw-r--r-- | 2,842 | weighted_sample_op.cu |
|
- | rw-r--r-- | 739 | weighted_sample_op.h |
|
- | rw-r--r-- | 841 | while_op.cc |
|
- | rw-r--r-- | 1,961 | while_op.h |
|
- | rw-r--r-- | 174 | while_op_gpu.cc |
|
- | rw-r--r-- | 1,267 | workspace_ops.cc |
|
- | rw-r--r-- | 797 | zero_gradient_op.cc |
|
- | rw-r--r-- | 347 | zero_gradient_op.h |
|
- | rw-r--r-- | 173 | zero_gradient_op_gpu.cc |