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,506 | abs_op.cc |
|
- | rw-r--r-- | 1,404 | 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,613 | accuracy_op.cc |
|
- | rw-r--r-- | 2,287 | accuracy_op.cu |
|
- | rw-r--r-- | 652 | accuracy_op.h |
|
- | rw-r--r-- | 1,869 | acos_op.cc |
|
- | rw-r--r-- | 1,349 | acos_op.cu |
|
- | rw-r--r-- | 711 | acos_op.h |
|
- | rw-r--r-- | 4,122 | activation_ops_cudnn.h |
|
- | rw-r--r-- | 5,582 | affine_channel_op.cc |
|
- | rw-r--r-- | 4,452 | 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,747 | apmeter_op.cc |
|
- | rw-r--r-- | 1,027 | apmeter_op.h |
|
- | rw-r--r-- | 5,879 | arg_ops.cc |
|
- | rw-r--r-- | 2,754 | arg_ops.cu |
|
- | rw-r--r-- | 2,356 | arg_ops.h |
|
- | rw-r--r-- | 1,864 | asin_op.cc |
|
- | rw-r--r-- | 1,347 | 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,374 | assert_op.h |
|
- | rw-r--r-- | 1,582 | async_net_barrier_op.cc |
|
- | rw-r--r-- | 205 | async_net_barrier_op.cu |
|
- | rw-r--r-- | 904 | async_net_barrier_op.h |
|
- | rw-r--r-- | 1,862 | atan_op.cc |
|
- | rw-r--r-- | 1,340 | atan_op.cu |
|
- | rw-r--r-- | 711 | atan_op.h |
|
- | rw-r--r-- | 5,756 | atomic_ops.cc |
|
- | rw-r--r-- | 11,489 | 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,769 | batch_gather_ops.cu |
|
- | rw-r--r-- | 5,275 | batch_gather_ops.h |
|
- | rw-r--r-- | 8,749 | batch_matmul_op.cc |
|
- | rw-r--r-- | 666 | 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,530 | batch_matmul_op_test.cc |
|
- | rw-r--r-- | 3,603 | batch_moments_op.cc |
|
- | rw-r--r-- | 4,294 | batch_moments_op.cu |
|
- | rw-r--r-- | 3,364 | batch_moments_op.h |
|
- | rw-r--r-- | 4,713 | batch_permutation_op.cc |
|
- | rw-r--r-- | 3,019 | 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-- | 7,406 | batch_sparse_to_dense_op.cc |
|
- | rw-r--r-- | 4,348 | batch_sparse_to_dense_op.cu |
|
- | rw-r--r-- | 6,147 | batch_sparse_to_dense_op.h |
|
- | rw-r--r-- | 7,662 | 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-- | 5,023 | bisect_percentile_op.h |
|
- | rw-r--r-- | 18,284 | boolean_mask_ops.cc |
|
- | rw-r--r-- | 12,054 | 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,509 | boolean_unmask_ops.cu |
|
- | rw-r--r-- | 378 | boolean_unmask_ops.h |
|
- | rw-r--r-- | 1,655 | boolean_unmask_ops_test.cc |
|
- | rw-r--r-- | 11,644 | box_with_nms_limit_op.cc |
|
- | rw-r--r-- | 4,960 | box_with_nms_limit_op.h |
|
- | rw-r--r-- | 2,324 | bucketize_op.cc |
|
- | rw-r--r-- | 1,439 | bucketize_op.cu |
|
- | rw-r--r-- | 1,361 | bucketize_op.h |
|
- | rw-r--r-- | 268 | byte_weight_dequant_op.cc |
|
- | rw-r--r-- | 1,732 | byte_weight_dequant_op.h |
|
- | rw-r--r-- | 6,592 | cast_op.cc |
|
- | rw-r--r-- | 3,769 | cast_op.cu |
|
- | rw-r--r-- | 1,350 | cast_op.h |
|
- | rw-r--r-- | 1,884 | cbrt_op.cc |
|
- | rw-r--r-- | 1,342 | cbrt_op.cu |
|
- | rw-r--r-- | 723 | cbrt_op.h |
|
- | rw-r--r-- | 274 | cc_bmm_bg_op.cc |
|
- | rw-r--r-- | 3,980 | cc_bmm_bg_op.h |
|
- | rw-r--r-- | 1,612 | ceil_op.cc |
|
- | rw-r--r-- | 772 | ceil_op.cu |
|
- | rw-r--r-- | 837 | ceil_op.h |
|
- | rw-r--r-- | 2,987 | channel_backprop_stats_op.cc |
|
- | rw-r--r-- | 6,165 | 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,673 | 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,988 | channel_stats_op.cu |
|
- | rw-r--r-- | 1,807 | channel_stats_op.h |
|
- | rw-r--r-- | 21,341 | channelwise_conv3d_op_cudnn.cu |
|
- | rw-r--r-- | 3,436 | clip_op.cc |
|
- | rw-r--r-- | 2,294 | clip_op.cu |
|
- | rw-r--r-- | 1,639 | clip_op.h |
|
- | rw-r--r-- | 23,414 | 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-- | 19,197 | concat_split_op.cc |
|
- | rw-r--r-- | 12,369 | 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,907 | conv_gradient_op.cc |
|
- | rw-r--r-- | 7,593 | 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,935 | conv_op_cudnn.cc |
|
- | rw-r--r-- | 8,704 | conv_op_eigen.cc |
|
- | rw-r--r-- | 704 | conv_op_gpu.cc |
|
- | rw-r--r-- | 28,916 | conv_op_impl.h |
|
- | rw-r--r-- | 1,097 | conv_op_shared.cc |
|
- | rw-r--r-- | 672 | conv_op_shared.h |
|
- | rw-r--r-- | 910 | conv_op_shared_gpu.cc |
|
- | rw-r--r-- | 32,166 | conv_pool_op_base.h |
|
- | rw-r--r-- | 1,648 | conv_transpose_gradient_op.cc |
|
- | rw-r--r-- | 4,474 | 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,222 | 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,631 | conv_transpose_op_mobile_impl.h |
|
- | rw-r--r-- | 6,682 | conv_transpose_op_mobile_test.cc |
|
- | rw-r--r-- | 10,307 | conv_transpose_unpool_op_base.h |
|
- | rw-r--r-- | 6,785 | 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,599 | copy_rows_to_tensor_op.h |
|
- | rw-r--r-- | 2,432 | cos_op.cc |
|
- | rw-r--r-- | 1,292 | cos_op.cu |
|
- | rw-r--r-- | 705 | cos_op.h |
|
- | rw-r--r-- | 2,478 | cosh_op.cc |
|
- | rw-r--r-- | 1,316 | cosh_op.cu |
|
- | rw-r--r-- | 711 | cosh_op.h |
|
- | rw-r--r-- | 2,570 | cosine_embedding_criterion_op.cc |
|
- | rw-r--r-- | 2,151 | cosine_embedding_criterion_op.cu |
|
- | rw-r--r-- | 1,127 | cosine_embedding_criterion_op.h |
|
- | rw-r--r-- | 9,136 | counter_ops.cc |
|
- | rw-r--r-- | 4,596 | 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,232 | create_scope_op.h |
|
- | rw-r--r-- | 7,305 | crf_viterbi_op.cc |
|
- | rw-r--r-- | 23,600 | cross_entropy_op.cc |
|
- | rw-r--r-- | 13,973 | cross_entropy_op.cu |
|
- | rw-r--r-- | 4,420 | cross_entropy_op.h |
|
- | rw-r--r-- | 6,800 | ctc_beam_search_decoder_op.cc |
|
- | rw-r--r-- | 1,102 | ctc_beam_search_decoder_op.h |
|
- | rw-r--r-- | 3,130 | ctc_greedy_decoder_op.cc |
|
- | rw-r--r-- | 817 | ctc_greedy_decoder_op.h |
|
- | rw-r--r-- | 1,783 | cube_op.cc |
|
- | rw-r--r-- | 1,301 | 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-- | 53,888 | dataset_ops.cc |
|
- | rw-r--r-- | 5,501 | dataset_ops.h |
|
- | rw-r--r-- | 1,780 | deform_conv_gradient_op.cc |
|
- | rw-r--r-- | 2,371 | deform_conv_op.cc |
|
- | rw-r--r-- | 21,718 | deform_conv_op.cu |
|
- | rw-r--r-- | 3,543 | deform_conv_op.h |
|
- | rw-r--r-- | 13,323 | deform_conv_op_impl.h |
|
- | rw-r--r-- | 1,182 | dense_vector_to_id_list_op.cc |
|
- | rw-r--r-- | 1,841 | dense_vector_to_id_list_op.h |
|
- | rw-r--r-- | 17,237 | depthwise_3x3_conv_op_cudnn.cu |
|
- | rw-r--r-- | 23,992 | distance_op.cc |
|
- | rw-r--r-- | 15,634 | distance_op.cu |
|
- | rw-r--r-- | 8,472 | distance_op.h |
|
- | rw-r--r-- | 1,193 | do_op.cc |
|
- | rw-r--r-- | 7,007 | do_op.h |
|
- | rw-r--r-- | 165 | do_op_gpu.cc |
|
- | rw-r--r-- | 6,080 | dropout_op.cc |
|
- | rw-r--r-- | 2,734 | dropout_op.cu |
|
- | rw-r--r-- | 1,450 | 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,820 | elementwise_add_op.h |
|
- | rw-r--r-- | 394 | elementwise_add_op_gpu.cc |
|
- | rw-r--r-- | 9,614 | elementwise_div_gradient_op.cc |
|
- | rw-r--r-- | 202 | elementwise_div_op.cc |
|
- | rw-r--r-- | 14,114 | 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,562 | 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,107 | elementwise_logical_ops.h |
|
- | rw-r--r-- | 10,383 | elementwise_mul_gradient_op.cc |
|
- | rw-r--r-- | 201 | elementwise_mul_op.cc |
|
- | rw-r--r-- | 7,850 | 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,287 | elementwise_op_test.h |
|
- | rw-r--r-- | 3,809 | elementwise_ops.cc |
|
- | rw-r--r-- | 7,687 | elementwise_ops.cu |
|
- | rw-r--r-- | 19,115 | elementwise_ops.h |
|
- | rw-r--r-- | 24,740 | elementwise_ops_schema.cc |
|
- | rw-r--r-- | 3,838 | 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,821 | 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,865 | elu_op.cc |
|
- | rw-r--r-- | 2,222 | 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-- | 2,339 | 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,863 | erf_op.cc |
|
- | rw-r--r-- | 1,361 | 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,914 | 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,430 | expand_squeeze_dims_op.h |
|
- | rw-r--r-- | 252 | expand_squeeze_dims_op_gpu.cc |
|
- | rw-r--r-- | 4,973 | fc_inference.cc |
|
- | rw-r--r-- | 775 | fc_inference.h |
|
- | rw-r--r-- | 14,801 | feature_maps_ops.cc |
|
- | rw-r--r-- | 31,935 | feature_maps_ops.h |
|
- | rw-r--r-- | 508 | feed_blob_op.cc |
|
- | rw-r--r-- | 802 | feed_blob_op.h |
|
- | rw-r--r-- | 21,580 | filler_op.cc |
|
- | rw-r--r-- | 2,226 | filler_op.cu |
|
- | rw-r--r-- | 18,474 | filler_op.h |
|
- | rw-r--r-- | 1,509 | find_duplicate_elements_op.cc |
|
- | rw-r--r-- | 1,597 | find_duplicate_elements_op.h |
|
- | rw-r--r-- | 739 | find_op.cc |
|
- | rw-r--r-- | 1,560 | find_op.cu |
|
- | rw-r--r-- | 2,108 | find_op.h |
|
- | rw-r--r-- | 2,178 | flatten_op.cc |
|
- | rw-r--r-- | 1,525 | flatten_op.h |
|
- | rw-r--r-- | 5,991 | flexible_top_k.cc |
|
- | rw-r--r-- | 936 | flexible_top_k.h |
|
- | rw-r--r-- | 1,719 | floor_op.cc |
|
- | rw-r--r-- | 780 | floor_op.cu |
|
- | rw-r--r-- | 825 | 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,456 | fully_connected_op.cc |
|
- | rw-r--r-- | 9,357 | fully_connected_op.h |
|
- | rw-r--r-- | 6,047 | fully_connected_op_gpu.cc |
|
- | rw-r--r-- | 12,811 | fused_rowwise_8bit_conversion_ops.cc |
|
- | rw-r--r-- | 6,632 | fused_rowwise_8bit_conversion_ops.h |
|
- | rw-r--r-- | 11,672 | fused_rowwise_nbit_conversion_ops.cc |
|
- | rw-r--r-- | 8,732 | fused_rowwise_nbit_conversion_ops.h |
|
- | rw-r--r-- | 8,529 | fused_rowwise_nbitfake_conversion_ops.cc |
|
- | rw-r--r-- | 4,346 | fused_rowwise_nbitfake_conversion_ops.h |
|
- | rw-r--r-- | 8,418 | 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,156 | 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,897 | gather_op.cuh |
|
- | rw-r--r-- | 7,560 | gather_op.h |
|
- | rw-r--r-- | 3,903 | gather_ranges_to_dense_op.cc |
|
- | rw-r--r-- | 8,426 | gather_ranges_to_dense_op.h |
|
- | rw-r--r-- | 3,780 | gelu_op.cc |
|
- | rw-r--r-- | 6,937 | gelu_op.cu |
|
- | rw-r--r-- | 1,452 | gelu_op.h |
|
- | rw-r--r-- | 16,347 | generate_proposals_op.cc |
|
- | rw-r--r-- | 24,293 | generate_proposals_op.cu |
|
- | rw-r--r-- | 6,262 | generate_proposals_op.h |
|
- | rw-r--r-- | 29,988 | generate_proposals_op_gpu_test.cc |
|
- | rw-r--r-- | 33,213 | generate_proposals_op_test.cc |
|
- | rw-r--r-- | 14,286 | generate_proposals_op_util_boxes.h |
|
- | rw-r--r-- | 5,654 | generate_proposals_op_util_boxes_test.cc |
|
- | rw-r--r-- | 26,113 | generate_proposals_op_util_nms.h |
|
- | rw-r--r-- | 18,963 | generate_proposals_op_util_nms_gpu.cu |
|
- | rw-r--r-- | 2,128 | generate_proposals_op_util_nms_gpu.h |
|
- | rw-r--r-- | 22,491 | generate_proposals_op_util_nms_gpu_test.cc |
|
- | rw-r--r-- | 18,987 | 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,146 | 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-- | 3,010 | given_tensor_fill_op.h |
|
- | rw-r--r-- | 1,616 | glu_op.cc |
|
- | rw-r--r-- | 1,176 | glu_op.cu |
|
- | rw-r--r-- | 1,458 | glu_op.h |
|
- | rw-r--r-- | 14,608 | group_norm_op.cc |
|
- | rw-r--r-- | 16,802 | 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,687 | gru_unit_op.h |
|
- | rw-r--r-- | 3,640 | gru_unit_op_gpu.cu |
|
- | rw-r--r-- | 23,595 | h_softmax_op.cc |
|
- | rw-r--r-- | 4,983 | h_softmax_op.h |
|
- | rw-r--r-- | 5,871 | half_float_ops.cc |
|
- | rw-r--r-- | 2,524 | half_float_ops.cu |
|
- | rw-r--r-- | 2,732 | half_float_ops.h |
|
- | rw-r--r-- | 3,401 | half_float_ops_test.cc |
|
- | rw-r--r-- | 3,912 | hard_sigmoid_op.cc |
|
- | rw-r--r-- | 2,251 | hard_sigmoid_op.cu |
|
- | rw-r--r-- | 994 | hard_sigmoid_op.h |
|
- | rw-r--r-- | 6,497 | 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,455 | 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-- | 9,004 | im2col_op.h |
|
- | rw-r--r-- | 245 | im2col_op_gpu.cc |
|
- | rw-r--r-- | 1,364 | index_hash_ops.cc |
|
- | rw-r--r-- | 2,268 | index_hash_ops.h |
|
- | rw-r--r-- | 11,692 | index_ops.cc |
|
- | rw-r--r-- | 3,212 | index_ops.h |
|
- | rw-r--r-- | 1,918 | inference_lstm_op.cc |
|
- | rw-r--r-- | 9,923 | inference_lstm_op.h |
|
- | rw-r--r-- | 8,018 | instance_norm_gradient_op.cc |
|
- | rw-r--r-- | 6,541 | instance_norm_op.cc |
|
- | rw-r--r-- | 13,741 | 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,948 | 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,940 | jsd_op.cc |
|
- | rw-r--r-- | 721 | jsd_op.h |
|
- | rw-r--r-- | 308 | key_split_ops.cc |
|
- | rw-r--r-- | 1,494 | key_split_ops.h |
|
- | rw-r--r-- | 7,083 | last_n_window_collector.cc |
|
- | rw-r--r-- | 9,489 | layer_norm_op.cc |
|
- | rw-r--r-- | 10,793 | layer_norm_op.cu |
|
- | rw-r--r-- | 8,012 | layer_norm_op.h |
|
- | rw-r--r-- | 3,440 | leaky_relu_op.cc |
|
- | rw-r--r-- | 1,737 | 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,304 | 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,607 | lengths_pad_op.h |
|
- | rw-r--r-- | 3,974 | lengths_reducer_fused_8bit_rowwise_ops.cc |
|
- | rw-r--r-- | 5,540 | lengths_reducer_fused_8bit_rowwise_ops.h |
|
- | rw-r--r-- | 22,337 | lengths_reducer_fused_nbit_rowwise_ops.cc |
|
- | rw-r--r-- | 23,598 | lengths_reducer_fused_nbit_rowwise_ops.h |
|
- | rw-r--r-- | 7,879 | lengths_reducer_ops.cc |
|
- | rw-r--r-- | 23,469 | lengths_reducer_ops.h |
|
- | rw-r--r-- | 8,383 | lengths_reducer_rowwise_8bit_ops.cc |
|
- | rw-r--r-- | 6,137 | lengths_reducer_rowwise_8bit_ops.h |
|
- | rw-r--r-- | 2,821 | lengths_tile_op.cc |
|
- | rw-r--r-- | 3,407 | lengths_tile_op.cu |
|
- | rw-r--r-- | 582 | lengths_tile_op.h |
|
- | rw-r--r-- | 5,343 | lengths_top_k_op.cc |
|
- | rw-r--r-- | 1,358 | lengths_top_k_op.h |
|
- | rw-r--r-- | 8,930 | listwise_l2r_op.cc |
|
- | rw-r--r-- | 1,677 | listwise_l2r_op.h |
|
- | rw-r--r-- | 18,174 | load_save_op.cc |
|
- | rw-r--r-- | 14,137 | 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,642 | load_save_op_util.h |
|
- | rw-r--r-- | 15,928 | local_response_normalization_op.cc |
|
- | rw-r--r-- | 11,756 | local_response_normalization_op.cu |
|
- | rw-r--r-- | 2,828 | local_response_normalization_op.h |
|
- | rw-r--r-- | 5,748 | local_response_normalization_op_cudnn.cc |
|
- | rw-r--r-- | 5,496 | 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,717 | locally_connected_op_impl.h |
|
- | rw-r--r-- | 2,108 | locally_connected_op_util.cc |
|
- | rw-r--r-- | 1,332 | locally_connected_op_util.h |
|
- | rw-r--r-- | 2,038 | log1p_op.cc |
|
- | rw-r--r-- | 1,326 | log1p_op.cu |
|
- | rw-r--r-- | 717 | log1p_op.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,738 | 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,142 | lp_pool_op.cu |
|
- | rw-r--r-- | 5,122 | 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,283 | lstm_unit_op_gpu.cu |
|
- | rw-r--r-- | 9,536 | lstm_utils.h |
|
- | rw-r--r-- | 2,494 | map_ops.cc |
|
- | rw-r--r-- | 8,056 | map_ops.h |
|
- | rw-r--r-- | 5,628 | margin_loss_l2r_op.cc |
|
- | rw-r--r-- | 1,324 | margin_loss_l2r_op.h |
|
- | rw-r--r-- | 3,901 | margin_ranking_criterion_op.cc |
|
- | rw-r--r-- | 2,703 | margin_ranking_criterion_op.cu |
|
- | rw-r--r-- | 1,113 | margin_ranking_criterion_op.h |
|
- | rw-r--r-- | 7,301 | matmul_op.cc |
|
- | rw-r--r-- | 2,843 | matmul_op.h |
|
- | rw-r--r-- | 164 | matmul_op_gpu.cc |
|
- | rw-r--r-- | 7,754 | 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,314 | mean_op.h |
|
- | rw-r--r-- | 1,566 | mem_query_op.cu |
|
- | rw-r--r-- | 1,637 | merge_id_lists_op.cc |
|
- | rw-r--r-- | 2,596 | 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,612 | minmax_ops.cu |
|
- | rw-r--r-- | 3,880 | minmax_ops.h |
|
- | rw-r--r-- | 2,874 | mish_op.cc |
|
- | rw-r--r-- | 794 | mish_op.h |
|
- | rw-r--r-- | 2,248 | mod_op.cc |
|
- | rw-r--r-- | 1,556 | mod_op.cu |
|
- | rw-r--r-- | 984 | mod_op.h |
|
- | rw-r--r-- | 2,720 | moments_op.cc |
|
- | rw-r--r-- | 3,114 | moments_op.cu |
|
- | rw-r--r-- | 4,088 | moments_op.h |
|
- | rw-r--r-- | 2,415 | multi_class_accuracy_op.cc |
|
- | rw-r--r-- | 2,308 | 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,705 | ngram_ops.h |
|
- | rw-r--r-- | 1,063 | no_default_engine_op.h |
|
- | rw-r--r-- | 1,765 | 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,021 | normalize_op.h |
|
- | rw-r--r-- | 5,154 | normalize_ops.cu |
|
- | rw-r--r-- | 503 | numpy_tile_op.cc |
|
- | rw-r--r-- | 3,800 | numpy_tile_op.h |
|
- | rw-r--r-- | 12,539 | one_hot_ops.cc |
|
- | rw-r--r-- | 1,005 | 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,724 | onnx_while_op.h |
|
- | rw-r--r-- | 2,120 | op_utils_cudnn.h |
|
- | rw-r--r-- | 4,183 | 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,189 | 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,112 | pack_rnn_sequence_op.h |
|
- | rw-r--r-- | 8,676 | pack_segments.cc |
|
- | rw-r--r-- | 9,929 | pack_segments.cu |
|
- | rw-r--r-- | 2,729 | pack_segments.h |
|
- | rw-r--r-- | 15,474 | pad_op.cc |
|
- | rw-r--r-- | 2,902 | pad_op.h |
|
- | rw-r--r-- | 16,557 | pad_op_gpu.cu |
|
- | rw-r--r-- | 5,565 | partition_ops.cc |
|
- | rw-r--r-- | 10,081 | partition_ops.h |
|
- | rw-r--r-- | 6,216 | percentile_op.cc |
|
- | rw-r--r-- | 1,009 | percentile_op.h |
|
- | rw-r--r-- | 1,265 | perplexity_op.cc |
|
- | rw-r--r-- | 1,196 | perplexity_op.cu |
|
- | rw-r--r-- | 447 | perplexity_op.h |
|
- | rw-r--r-- | 3,945 | piecewise_linear_transform_op.cc |
|
- | rw-r--r-- | 8,048 | piecewise_linear_transform_op.cu |
|
- | rw-r--r-- | 8,407 | piecewise_linear_transform_op.h |
|
- | rw-r--r-- | 26,669 | pool_gradient_op.cc |
|
- | rw-r--r-- | 42,714 | pool_op.cc |
|
- | rw-r--r-- | 61,213 | pool_op.cu |
|
- | rw-r--r-- | 8,541 | 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,057 | pow_op.cc |
|
- | rw-r--r-- | 2,768 | pow_op.cu |
|
- | rw-r--r-- | 4,677 | pow_op.h |
|
- | rw-r--r-- | 4,661 | prefetch_op.h |
|
- | rw-r--r-- | 10,552 | prelu_op.cc |
|
- | rw-r--r-- | 7,817 | prelu_op.cu |
|
- | rw-r--r-- | 1,067 | prelu_op.h |
|
- | rw-r--r-- | 1,342 | prepend_dim_op.cc |
|
- | rw-r--r-- | 2,800 | prepend_dim_op.h |
|
- | rw-r--r-- | 248 | prepend_dim_op_gpu.cc |
|
- | rw-r--r-- | 2,261 | quant_decode_op.cc |
|
- | rw-r--r-- | 5,447 | quant_decode_op.h |
|
- | rw-r--r-- | 614 | quantile_op.cc |
|
- | rw-r--r-- | 4,193 | quantile_op.h |
|
- | rw-r--r-- | 6,325 | rank_loss_op.cc |
|
- | rw-r--r-- | 820 | rank_loss_op.h |
|
- | rw-r--r-- | 1,371 | reciprocal_gradient_op.cc |
|
- | rw-r--r-- | 1,369 | reciprocal_op.cc |
|
- | rw-r--r-- | 1,392 | 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,941 | reduce_front_back_max_ops.cu |
|
- | rw-r--r-- | 4,439 | reduce_front_back_max_ops.h |
|
- | rw-r--r-- | 8,600 | reduce_front_back_mean_ops.cc |
|
- | rw-r--r-- | 7,361 | reduce_front_back_sum_mean_ops.cu |
|
- | rw-r--r-- | 5,377 | reduce_front_back_sum_mean_ops.h |
|
- | rw-r--r-- | 8,300 | reduce_front_back_sum_ops.cc |
|
- | rw-r--r-- | 15,128 | reduce_ops.cc |
|
- | rw-r--r-- | 4,542 | reduce_ops.cu |
|
- | rw-r--r-- | 9,999 | reduce_ops.h |
|
- | rw-r--r-- | 24,714 | reducer_functors.h |
|
- | rw-r--r-- | 9,855 | reduction_ops.cc |
|
- | rw-r--r-- | 3,868 | reduction_ops.cu |
|
- | rw-r--r-- | 5,981 | reduction_ops.h |
|
- | rw-r--r-- | 2,845 | relu_n_op.cc |
|
- | rw-r--r-- | 2,127 | relu_n_op.cu |
|
- | rw-r--r-- | 990 | relu_n_op.h |
|
- | rw-r--r-- | 4,055 | relu_op.cc |
|
- | rw-r--r-- | 7,219 | relu_op.cu |
|
- | rw-r--r-- | 624 | relu_op.h |
|
- | rw-r--r-- | 798 | remove_data_blocks_op.cc |
|
- | rw-r--r-- | 2,691 | remove_data_blocks_op.h |
|
- | rw-r--r-- | 854 | replace_nan_op.cc |
|
- | rw-r--r-- | 779 | replace_nan_op.cu |
|
- | rw-r--r-- | 1,170 | replace_nan_op.h |
|
- | rw-r--r-- | 9,087 | reservoir_sampling.cc |
|
- | rw-r--r-- | 5,831 | reshape_op.cc |
|
- | rw-r--r-- | 5,768 | 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,653 | resize_3d_op.cc |
|
- | rw-r--r-- | 6,489 | resize_3d_op.cu |
|
- | rw-r--r-- | 2,677 | resize_3d_op.h |
|
- | rw-r--r-- | 12,551 | resize_op.cc |
|
- | rw-r--r-- | 5,078 | 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,564 | reverse_packed_segs_op.cu |
|
- | rw-r--r-- | 2,805 | reverse_packed_segs_op.h |
|
- | rw-r--r-- | 4,338 | rmac_regions_op.cc |
|
- | rw-r--r-- | 6,235 | 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,349 | rms_norm_op.cu |
|
- | rw-r--r-- | 2,968 | rms_norm_op.h |
|
- | rw-r--r-- | 8,493 | roi_align_gradient_op.cc |
|
- | rw-r--r-- | 7,045 | roi_align_gradient_op.cu |
|
- | rw-r--r-- | 1,510 | roi_align_gradient_op.h |
|
- | rw-r--r-- | 10,765 | roi_align_op.cc |
|
- | rw-r--r-- | 5,540 | roi_align_op.cu |
|
- | rw-r--r-- | 2,875 | 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-- | 6,941 | roi_align_rotated_gradient_op.cu |
|
- | rw-r--r-- | 1,393 | roi_align_rotated_gradient_op.h |
|
- | rw-r--r-- | 13,627 | roi_align_rotated_op.cc |
|
- | rw-r--r-- | 6,097 | roi_align_rotated_op.cu |
|
- | rw-r--r-- | 1,660 | roi_align_rotated_op.h |
|
- | rw-r--r-- | 7,839 | roi_pool_op.cc |
|
- | rw-r--r-- | 6,739 | roi_pool_op.cu |
|
- | rw-r--r-- | 2,503 | roi_pool_op.h |
|
- | rw-r--r-- | 1,550 | rowmul_op.cc |
|
- | rw-r--r-- | 2,008 | rowmul_op.h |
|
- | rw-r--r-- | 1,818 | rsqrt_op.cc |
|
- | rw-r--r-- | 1,445 | rsqrt_op.cu |
|
- | rw-r--r-- | 729 | rsqrt_op.h |
|
- | rw-r--r-- | 553 | scale_blobs_op.cc |
|
- | rw-r--r-- | 5,381 | scale_blobs_op.cu |
|
- | rw-r--r-- | 1,503 | 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-- | 23,912 | segment_reduction_op.cc |
|
- | rw-r--r-- | 71,035 | segment_reduction_op.h |
|
- | rw-r--r-- | 58,597 | segment_reduction_op_gpu.cu |
|
- | rw-r--r-- | 2,831 | segment_reduction_op_gpu.cuh |
|
- | rw-r--r-- | 1,823 | self_binning_histogram_op.cc |
|
- | rw-r--r-- | 6,289 | self_binning_histogram_op.h |
|
- | rw-r--r-- | 4,131 | selu_op.cc |
|
- | rw-r--r-- | 1,954 | selu_op.cu |
|
- | rw-r--r-- | 1,545 | selu_op.h |
|
- | rw-r--r-- | 17,471 | sequence_ops.cc |
|
- | rw-r--r-- | 11,023 | sequence_ops.cu |
|
- | rw-r--r-- | 8,264 | sequence_ops.h |
|
- | rw-r--r-- | 1,769 | shape_op.cc |
|
- | rw-r--r-- | 1,675 | shape_op.h |
|
- | rw-r--r-- | 151 | shape_op_gpu.cc |
|
- | rw-r--r-- | 1,357 | sigmoid_gradient_op.cc |
|
- | rw-r--r-- | 1,997 | sigmoid_op.cc |
|
- | rw-r--r-- | 2,047 | sigmoid_op.cu |
|
- | rw-r--r-- | 639 | sigmoid_op.h |
|
- | rw-r--r-- | 320 | sigmoid_op_cudnn.cc |
|
- | rw-r--r-- | 2,406 | sin_op.cc |
|
- | rw-r--r-- | 1,302 | sin_op.cu |
|
- | rw-r--r-- | 705 | sin_op.h |
|
- | rw-r--r-- | 2,474 | sinh_op.cc |
|
- | rw-r--r-- | 1,309 | sinh_op.cu |
|
- | rw-r--r-- | 711 | sinh_op.h |
|
- | rw-r--r-- | 1,404 | sinusoid_position_encoding_op.cc |
|
- | rw-r--r-- | 2,853 | sinusoid_position_encoding_op.h |
|
- | rw-r--r-- | 4,145 | slice_op.cc |
|
- | rw-r--r-- | 10,974 | slice_op.cu |
|
- | rw-r--r-- | 10,150 | slice_op.h |
|
- | rw-r--r-- | 5,772 | softmax_op.cc |
|
- | rw-r--r-- | 1,174 | softmax_op.h |
|
- | rw-r--r-- | 4,054 | softmax_op_cudnn.cc |
|
- | rw-r--r-- | 24,269 | softmax_ops.cu |
|
- | rw-r--r-- | 1,867 | softmax_utils.cc |
|
- | rw-r--r-- | 447 | softmax_utils.h |
|
- | rw-r--r-- | 13,453 | softmax_with_loss_op.cc |
|
- | rw-r--r-- | 2,883 | softmax_with_loss_op.h |
|
- | rw-r--r-- | 2,983 | softplus_op.cc |
|
- | rw-r--r-- | 1,697 | softplus_op.cu |
|
- | rw-r--r-- | 781 | softplus_op.h |
|
- | rw-r--r-- | 3,683 | softsign_op.cc |
|
- | rw-r--r-- | 2,101 | softsign_op.cu |
|
- | rw-r--r-- | 675 | softsign_op.h |
|
- | rw-r--r-- | 4,013 | space_batch_op.cc |
|
- | rw-r--r-- | 6,899 | space_batch_op.h |
|
- | rw-r--r-- | 4,920 | space_batch_op_gpu.cu |
|
- | rw-r--r-- | 3,935 | sparse_dropout_with_replacement_op.cc |
|
- | rw-r--r-- | 1,122 | sparse_dropout_with_replacement_op.h |
|
- | rw-r--r-- | 3,684 | sparse_itemwise_dropout_with_replacement_op.cc |
|
- | rw-r--r-- | 1,163 | sparse_itemwise_dropout_with_replacement_op.h |
|
- | rw-r--r-- | 3,537 | sparse_lp_regularizer_op.cc |
|
- | rw-r--r-- | 1,130 | sparse_lp_regularizer_op.h |
|
- | rw-r--r-- | 226 | sparse_lp_regularizer_op_gpu.cu |
|
- | rw-r--r-- | 5,603 | sparse_normalize_op.cc |
|
- | rw-r--r-- | 834 | sparse_normalize_op.h |
|
- | rw-r--r-- | 226 | sparse_normalize_op_gpu.cu |
|
- | rw-r--r-- | 4,903 | sparse_to_dense_mask_op.cc |
|
- | rw-r--r-- | 10,051 | sparse_to_dense_mask_op.h |
|
- | rw-r--r-- | 2,394 | sparse_to_dense_op.cc |
|
- | rw-r--r-- | 2,467 | sparse_to_dense_op.cu |
|
- | rw-r--r-- | 4,010 | sparse_to_dense_op.h |
|
- | rw-r--r-- | 6,326 | spatial_batch_norm_gradient_op.cc |
|
- | rw-r--r-- | 6,059 | 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-- | 16,327 | spatial_batch_norm_op_cudnn.cu |
|
- | rw-r--r-- | 12,398 | spatial_batch_norm_op_impl.cuh |
|
- | rw-r--r-- | 8,670 | 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,896 | square_root_divide_op.h |
|
- | rw-r--r-- | 10,149 | stats_ops.cc |
|
- | rw-r--r-- | 3,912 | stats_put_ops.cc |
|
- | rw-r--r-- | 2,813 | 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-- | 7,068 | string_ops.cc |
|
- | rw-r--r-- | 2,075 | string_ops.h |
|
- | rw-r--r-- | 4,544 | string_ops_test.cc |
|
- | rw-r--r-- | 3,038 | stump_func_op.cc |
|
- | rw-r--r-- | 1,624 | stump_func_op.cu |
|
- | rw-r--r-- | 2,112 | stump_func_op.h |
|
- | rw-r--r-- | 19,473 | stylizer_ops.cc |
|
- | rw-r--r-- | 2,377 | summarize_op.cc |
|
- | rw-r--r-- | 3,392 | summarize_op.cu |
|
- | rw-r--r-- | 1,875 | summarize_op.h |
|
- | rw-r--r-- | 2,662 | swish_op.cc |
|
- | rw-r--r-- | 2,198 | swish_op.cu |
|
- | rw-r--r-- | 772 | swish_op.h |
|
- | rw-r--r-- | 1,802 | tan_op.cc |
|
- | rw-r--r-- | 1,380 | tan_op.cu |
|
- | rw-r--r-- | 705 | tan_op.h |
|
- | rw-r--r-- | 1,346 | tanh_gradient_op.cc |
|
- | rw-r--r-- | 1,900 | tanh_op.cc |
|
- | rw-r--r-- | 1,340 | 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,688 | tensor_protos_db_input.h |
|
- | rw-r--r-- | 248 | tensor_protos_db_input_gpu.cc |
|
- | rw-r--r-- | 6,700 | text_file_reader.cc |
|
- | rw-r--r-- | 3,457 | text_file_reader_utils.cc |
|
- | rw-r--r-- | 2,913 | text_file_reader_utils.h |
|
- | rw-r--r-- | 4,550 | text_file_reader_utils_test.cc |
|
- | rw-r--r-- | 2,902 | thresholded_relu_op.cc |
|
- | rw-r--r-- | 1,712 | thresholded_relu_op.cu |
|
- | rw-r--r-- | 1,137 | thresholded_relu_op.h |
|
- | rw-r--r-- | 6,110 | tile_op.cc |
|
- | rw-r--r-- | 2,746 | tile_op.cu |
|
- | rw-r--r-- | 8,776 | tile_op.h |
|
- | rw-r--r-- | 10,664 | top_k.cc |
|
- | rw-r--r-- | 10,934 | top_k.cu |
|
- | rw-r--r-- | 1,061 | top_k.h |
|
- | rw-r--r-- | 8,908 | top_k_heap_selection.cuh |
|
- | rw-r--r-- | 15,577 | top_k_radix_selection.cuh |
|
- | rw-r--r-- | 3,545 | transpose_op.cc |
|
- | rw-r--r-- | 186 | transpose_op.cu |
|
- | rw-r--r-- | 2,086 | 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,456 | tt_linear_op.h |
|
- | rw-r--r-- | 3,593 | unique_ops.cc |
|
- | rw-r--r-- | 4,316 | unique_ops.cu |
|
- | rw-r--r-- | 1,666 | unique_ops.h |
|
- | rw-r--r-- | 1,090 | unsafe_coalesce.cc |
|
- | rw-r--r-- | 178 | unsafe_coalesce.cu |
|
- | rw-r--r-- | 2,533 | unsafe_coalesce.h |
|
- | rw-r--r-- | 8,686 | upsample_op.cc |
|
- | rw-r--r-- | 7,338 | upsample_op.cu |
|
- | rw-r--r-- | 2,246 | upsample_op.h |
|
- | rw-r--r-- | 34,672 | utility_ops.cc |
|
- | rw-r--r-- | 10,859 | utility_ops.cu |
|
- | rw-r--r-- | 49,935 | 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-- | 710 | variable_length_sequence_padding.cc |
|
- | rw-r--r-- | 1,386 | variable_length_sequence_padding.h |
|
- | rw-r--r-- | 4,090 | weighted_multi_sampling_op.cc |
|
- | rw-r--r-- | 602 | weighted_multi_sampling_op.h |
|
- | rw-r--r-- | 4,178 | weighted_sample_op.cc |
|
- | rw-r--r-- | 2,878 | 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 |