operators « caffe2 - platform/upstream/pytorch - Domain: Machine Learning / ML Framework; Licenses: BSD-3-Clause;

Mode	Name	Size
-rw-r--r--	CMakeLists.txt	6461	log plain
-rw-r--r--	abs_op.cc	2423	log plain
-rw-r--r--	abs_op.cu	1369	log plain
-rw-r--r--	abs_op.h	705	log plain
-rw-r--r--	accumulate_op.cc	1210	log plain
-rw-r--r--	accumulate_op.cu	194	log plain
-rw-r--r--	accumulate_op.h	1058	log plain
-rw-r--r--	accuracy_op.cc	2521	log plain
-rw-r--r--	accuracy_op.cu	2133	log plain
-rw-r--r--	accuracy_op.h	646	log plain
-rw-r--r--	acos_op.cc	1809	log plain
-rw-r--r--	acos_op.cu	1314	log plain
-rw-r--r--	acos_op.h	711	log plain
-rw-r--r--	activation_ops_cudnn.h	4068	log plain
-rw-r--r--	adjust_batch_op.cc	1025	log plain
-rw-r--r--	adjust_batch_op.h	2503	log plain
-rw-r--r--	affine_channel_op.cc	5398	log plain
-rw-r--r--	affine_channel_op.cu	4358	log plain
-rw-r--r--	affine_channel_op.h	3410	log plain
-rw-r--r--	apmeter_op.cc	3619	log plain
-rw-r--r--	apmeter_op.h	1012	log plain
-rw-r--r--	arg_ops.cc	5879	log plain
-rw-r--r--	arg_ops.cu	2665	log plain
-rw-r--r--	arg_ops.h	2304	log plain
-rw-r--r--	asin_op.cc	1804	log plain
-rw-r--r--	asin_op.cu	1312	log plain
-rw-r--r--	asin_op.h	711	log plain
-rw-r--r--	assert_op.cc	1482	log plain
-rw-r--r--	assert_op.cu	176	log plain
-rw-r--r--	assert_op.h	1320	log plain
-rw-r--r--	atan_op.cc	1802	log plain
-rw-r--r--	atan_op.cu	1305	log plain
-rw-r--r--	atan_op.h	711	log plain
-rw-r--r--	atomic_ops.cc	4759	log plain
-rw-r--r--	batch_box_cox_op.cc	11237	log plain
-rw-r--r--	batch_box_cox_op.h	2172	log plain
-rw-r--r--	batch_bucketize_op.cc	4243	log plain
-rw-r--r--	batch_bucketize_op.h	705	log plain
-rw-r--r--	batch_gather_ops.cc	2035	log plain
-rw-r--r--	batch_gather_ops.cu	4584	log plain
-rw-r--r--	batch_gather_ops.h	4399	log plain
-rw-r--r--	batch_matmul_op.cc	8785	log plain
-rw-r--r--	batch_matmul_op.cu	666	log plain
-rw-r--r--	batch_matmul_op.h	9181	log plain
-rw-r--r--	batch_matmul_op_gpu_test.cc	2535	log plain
-rw-r--r--	batch_matmul_op_test.cc	2214	log plain
-rw-r--r--	batch_moments_op.cc	3603	log plain
-rw-r--r--	batch_moments_op.cu	4112	log plain
-rw-r--r--	batch_moments_op.h	3334	log plain
-rw-r--r--	batch_sparse_to_dense_op.cc	7617	log plain
-rw-r--r--	batch_sparse_to_dense_op.h	1264	log plain
-rw-r--r--	bbox_transform_op.cc	7029	log plain
-rw-r--r--	bbox_transform_op.h	2689	log plain
-rw-r--r--	bisect_percentile_op.cc	3605	log plain
-rw-r--r--	bisect_percentile_op.h	4906	log plain
-rw-r--r--	boolean_mask_ops.cc	16488	log plain
-rw-r--r--	boolean_mask_ops.cu	11692	log plain
-rw-r--r--	boolean_mask_ops.h	1980	log plain
-rw-r--r--	boolean_unmask_ops.cc	4326	log plain
-rw-r--r--	boolean_unmask_ops.cu	3405	log plain
-rw-r--r--	boolean_unmask_ops.h	378	log plain
-rw-r--r--	boolean_unmask_ops_test.cc	1649	log plain
-rw-r--r--	box_with_nms_limit_op.cc	10346	log plain
-rw-r--r--	box_with_nms_limit_op.h	2350	log plain
-rw-r--r--	byte_weight_dequant_op.cc	268	log plain
-rw-r--r--	byte_weight_dequant_op.h	1720	log plain
-rw-r--r--	cast_op.cc	6592	log plain
-rw-r--r--	cast_op.cu	3734	log plain
-rw-r--r--	cast_op.h	1385	log plain
-rw-r--r--	cbrt_op.cc	1816	log plain
-rw-r--r--	cbrt_op.cu	1307	log plain
-rw-r--r--	cbrt_op.h	723	log plain
-rw-r--r--	ceil_op.cc	1612	log plain
-rw-r--r--	ceil_op.cu	739	log plain
-rw-r--r--	ceil_op.h	782	log plain
-rw-r--r--	channel_backprop_stats_op.cc	2987	log plain
-rw-r--r--	channel_backprop_stats_op.cu	6085	log plain
-rw-r--r--	channel_backprop_stats_op.h	722	log plain
-rw-r--r--	channel_shuffle_op.cc	4374	log plain
-rw-r--r--	channel_shuffle_op.cu	6265	log plain
-rw-r--r--	channel_shuffle_op.h	1872	log plain
-rw-r--r--	channel_stats_op.cc	2014	log plain
-rw-r--r--	channel_stats_op.cu	5514	log plain
-rw-r--r--	channel_stats_op.h	696	log plain
-rw-r--r--	clip_op.cc	3342	log plain
-rw-r--r--	clip_op.cu	2228	log plain
-rw-r--r--	clip_op.h	1609	log plain
-rw-r--r--	collect_and_distribute_fpn_rpn_proposals_op.cc	12801	log plain
-rw-r--r--	collect_and_distribute_fpn_rpn_proposals_op.h	3924	log plain
-rw-r--r--	communicator_op.cc	7199	log plain
-rw-r--r--	communicator_op_gpu.cc	774	log plain
-rw-r--r--	concat_split_op.cc	12876	log plain
-rw-r--r--	concat_split_op.h	10839	log plain
-rw-r--r--	concat_split_op_gpu.cc	459	log plain
-rw-r--r--	conditional_op.cc	2180	log plain
-rw-r--r--	conditional_op.h	462	log plain
-rw-r--r--	conv_gradient_op.cc	4854	log plain
-rw-r--r--	conv_op.cc	7516	log plain
-rw-r--r--	conv_op.h	3107	log plain
-rw-r--r--	conv_op_cache_cudnn.cc	419	log plain
-rw-r--r--	conv_op_cache_cudnn.h	1935	log plain
-rw-r--r--	conv_op_cache_cudnn_test.cc	1571	log plain
-rw-r--r--	conv_op_cudnn.cc	50515	log plain
-rw-r--r--	conv_op_eigen.cc	8311	log plain
-rw-r--r--	conv_op_gpu.cc	704	log plain
-rw-r--r--	conv_op_impl.h	28669	log plain
-rw-r--r--	conv_op_shared.cc	1054	log plain
-rw-r--r--	conv_op_shared.h	672	log plain
-rw-r--r--	conv_op_shared_gpu.cc	910	log plain
-rw-r--r--	conv_pool_op_base.h	26426	log plain
-rw-r--r--	conv_transpose_gradient_op.cc	1648	log plain
-rw-r--r--	conv_transpose_op.cc	4418	log plain
-rw-r--r--	conv_transpose_op.h	1697	log plain
-rw-r--r--	conv_transpose_op_cudnn.cc	22161	log plain
-rw-r--r--	conv_transpose_op_gpu.cc	357	log plain
-rw-r--r--	conv_transpose_op_impl.h	17499	log plain
-rw-r--r--	conv_transpose_op_mobile.cc	428	log plain
-rw-r--r--	conv_transpose_op_mobile.h	1517	log plain
-rw-r--r--	conv_transpose_op_mobile_impl.h	19545	log plain
-rw-r--r--	conv_transpose_op_mobile_test.cc	6776	log plain
-rw-r--r--	conv_transpose_unpool_op_base.h	9586	log plain
-rw-r--r--	copy_op.cc	6420	log plain
-rw-r--r--	copy_op.cu	1562	log plain
-rw-r--r--	copy_op.h	1131	log plain
-rw-r--r--	cos_op.cc	2372	log plain
-rw-r--r--	cos_op.cu	1257	log plain
-rw-r--r--	cos_op.h	705	log plain
-rw-r--r--	cosh_op.cc	2410	log plain
-rw-r--r--	cosh_op.cu	1274	log plain
-rw-r--r--	cosh_op.h	711	log plain
-rw-r--r--	cosine_embedding_criterion_op.cc	2570	log plain
-rw-r--r--	cosine_embedding_criterion_op.cu	2085	log plain
-rw-r--r--	cosine_embedding_criterion_op.h	1061	log plain
-rw-r--r--	counter_ops.cc	8886	log plain
-rw-r--r--	counter_ops.h	4507	log plain
-rw-r--r--	counter_ops_gpu.cc	582	log plain
-rw-r--r--	create_scope_op.cc	1241	log plain
-rw-r--r--	create_scope_op.h	5183	log plain
-rw-r--r--	crf_viterbi_op.cc	7274	log plain
-rw-r--r--	cross_entropy_op.cc	22705	log plain
-rw-r--r--	cross_entropy_op.cu	13727	log plain
-rw-r--r--	cross_entropy_op.h	4463	log plain
-rw-r--r--	ctc_beam_search_decoder_op.cc	5739	log plain
-rw-r--r--	ctc_beam_search_decoder_op.h	909	log plain
-rw-r--r--	ctc_greedy_decoder_op.cc	3023	log plain
-rw-r--r--	ctc_greedy_decoder_op.h	802	log plain
-rw-r--r--	cube_op.cc	1723	log plain
-rw-r--r--	cube_op.cu	1266	log plain
-rw-r--r--	cube_op.h	723	log plain
-rw-r--r--	data_couple.cc	590	log plain
-rw-r--r--	data_couple.h	464	log plain
-rw-r--r--	data_couple_gpu.cu	164	log plain
-rw-r--r--	dataset_ops.cc	49961	log plain
-rw-r--r--	dataset_ops.h	5334	log plain
-rw-r--r--	deform_conv_gradient_op.cc	1725	log plain
-rw-r--r--	deform_conv_op.cc	2371	log plain
-rw-r--r--	deform_conv_op.cu	21554	log plain
-rw-r--r--	deform_conv_op.h	3516	log plain
-rw-r--r--	deform_conv_op_impl.h	12820	log plain
-rw-r--r--	dense_vector_to_id_list_op.cc	1182	log plain
-rw-r--r--	dense_vector_to_id_list_op.h	1797	log plain
-rw-r--r--	depthwise_3x3_conv_op_cudnn.cu	17046	log plain
-rw-r--r--	distance_op.cc	22631	log plain
-rw-r--r--	distance_op.cu	14967	log plain
-rw-r--r--	distance_op.h	8061	log plain
-rw-r--r--	do_op.cc	1193	log plain
-rw-r--r--	do_op.h	6909	log plain
-rw-r--r--	do_op_gpu.cc	165	log plain
-rw-r--r--	dropout_op.cc	5597	log plain
-rw-r--r--	dropout_op.cu	2660	log plain
-rw-r--r--	dropout_op.h	1486	log plain
-rw-r--r--	dropout_op_cudnn.cc	8566	log plain
-rw-r--r--	elementwise_add_gradient_op.cc	683	log plain
-rw-r--r--	elementwise_add_op.cc	201	log plain
-rw-r--r--	elementwise_add_op.h	1808	log plain
-rw-r--r--	elementwise_add_op_gpu.cc	394	log plain
-rw-r--r--	elementwise_div_gradient_op.cc	8402	log plain
-rw-r--r--	elementwise_div_op.cc	202	log plain
-rw-r--r--	elementwise_div_op.cu	13989	log plain
-rw-r--r--	elementwise_div_op.h	1062	log plain
-rw-r--r--	elementwise_linear_op.cc	5112	log plain
-rw-r--r--	elementwise_linear_op.cu	3458	log plain
-rw-r--r--	elementwise_linear_op.h	1141	log plain
-rw-r--r--	elementwise_logical_ops.cc	3042	log plain
-rw-r--r--	elementwise_logical_ops.h	5033	log plain
-rw-r--r--	elementwise_mul_gradient_op.cc	3056	log plain
-rw-r--r--	elementwise_mul_op.cc	201	log plain
-rw-r--r--	elementwise_mul_op.cu	5899	log plain
-rw-r--r--	elementwise_mul_op.h	1062	log plain
-rw-r--r--	elementwise_op_gpu_test.cc	975	log plain
-rw-r--r--	elementwise_op_test.cc	764	log plain
-rw-r--r--	elementwise_op_test.h	9237	log plain
-rw-r--r--	elementwise_ops.cc	3747	log plain
-rw-r--r--	elementwise_ops.cu	6720	log plain
-rw-r--r--	elementwise_ops.h	16663	log plain
-rw-r--r--	elementwise_ops_schema.cc	22711	log plain
-rw-r--r--	elementwise_ops_utils.cc	2937	log plain
-rw-r--r--	elementwise_ops_utils.h	812	log plain
-rw-r--r--	elementwise_sub_gradient_op.cc	683	log plain
-rw-r--r--	elementwise_sub_op.cc	201	log plain
-rw-r--r--	elementwise_sub_op.h	1809	log plain
-rw-r--r--	elementwise_sub_op_gpu.cc	394	log plain
-rw-r--r--	elementwise_sum_op.cc	2475	log plain
-rw-r--r--	elu_op.cc	3805	log plain
-rw-r--r--	elu_op.cu	2152	log plain
-rw-r--r--	elu_op.h	875	log plain
-rw-r--r--	elu_op_cudnn.cc	3041	log plain
-rw-r--r--	enforce_finite_op.cc	524	log plain
-rw-r--r--	enforce_finite_op.cu	374	log plain
-rw-r--r--	enforce_finite_op.h	1119	log plain
-rw-r--r--	ensure_clipped_op.cc	1680	log plain
-rw-r--r--	ensure_clipped_op.h	1593	log plain
-rw-r--r--	ensure_cpu_output_op.cc	1139	log plain
-rw-r--r--	ensure_cpu_output_op.cu	293	log plain
-rw-r--r--	ensure_cpu_output_op.h	1442	log plain
-rw-r--r--	erf_op.cc	1803	log plain
-rw-r--r--	erf_op.cu	1326	log plain
-rw-r--r--	erf_op.h	751	log plain
-rw-r--r--	exp_op.cc	1903	log plain
-rw-r--r--	exp_op.h	425	log plain
-rw-r--r--	exp_op_gpu.cc	260	log plain
-rw-r--r--	expand_op.cc	1689	log plain
-rw-r--r--	expand_op.h	3429	log plain
-rw-r--r--	expand_op_gpu.cc	411	log plain
-rw-r--r--	expand_squeeze_dims_op.cc	5867	log plain
-rw-r--r--	expand_squeeze_dims_op.h	3353	log plain
-rw-r--r--	expand_squeeze_dims_op_gpu.cc	252	log plain
d---------	experimental / c10	30	log plain
-rw-r--r--	fc_inference.cc	2148	log plain
-rw-r--r--	fc_inference.h	429	log plain
-rw-r--r--	feature_maps_ops.cc	14276	log plain
-rw-r--r--	feature_maps_ops.h	30007	log plain
-rw-r--r--	feed_blob_op.cc	508	log plain
-rw-r--r--	feed_blob_op.h	769	log plain
-rw-r--r--	filler_op.cc	21408	log plain
-rw-r--r--	filler_op.cu	2156	log plain
-rw-r--r--	filler_op.h	17860	log plain
-rw-r--r--	find_duplicate_elements_op.cc	1509	log plain
-rw-r--r--	find_duplicate_elements_op.h	1556	log plain
-rw-r--r--	find_op.cc	739	log plain
-rw-r--r--	find_op.cu	1483	log plain
-rw-r--r--	find_op.h	2040	log plain
-rw-r--r--	flatten_op.cc	2747	log plain
-rw-r--r--	flatten_op.h	948	log plain
-rw-r--r--	flexible_top_k.cc	5785	log plain
-rw-r--r--	flexible_top_k.h	906	log plain
-rw-r--r--	floor_op.cc	1719	log plain
-rw-r--r--	floor_op.cu	745	log plain
-rw-r--r--	floor_op.h	788	log plain
-rw-r--r--	free_op.cc	408	log plain
-rw-r--r--	free_op.h	738	log plain
-rw-r--r--	free_op_gpu.cc	168	log plain
-rw-r--r--	fully_connected_op.cc	10555	log plain
-rw-r--r--	fully_connected_op.h	9299	log plain
-rw-r--r--	fully_connected_op_gpu.cc	6048	log plain
-rw-r--r--	fused_rowwise_8bit_conversion_ops.cc	5993	log plain
-rw-r--r--	fused_rowwise_8bit_conversion_ops.h	5570	log plain
-rw-r--r--	fused_rowwise_random_quantization_ops.cc	8315	log plain
-rw-r--r--	fused_rowwise_random_quantization_ops.h	2605	log plain
-rw-r--r--	gather_fused_8bit_rowwise_op.cc	1272	log plain
-rw-r--r--	gather_fused_8bit_rowwise_op.h	2179	log plain
-rw-r--r--	gather_op.cc	4174	log plain
-rw-r--r--	gather_op.cu	632	log plain
-rw-r--r--	gather_op.cuh	3738	log plain
-rw-r--r--	gather_op.h	6628	log plain
-rw-r--r--	gather_ranges_to_dense_op.cc	2913	log plain
-rw-r--r--	gather_ranges_to_dense_op.h	4472	log plain
-rw-r--r--	generate_proposals_op.cc	15523	log plain
-rw-r--r--	generate_proposals_op.cu	23427	log plain
-rw-r--r--	generate_proposals_op.h	6271	log plain
-rw-r--r--	generate_proposals_op_gpu_test.cc	30026	log plain
-rw-r--r--	generate_proposals_op_test.cc	33201	log plain
-rw-r--r--	generate_proposals_op_util_boxes.h	13690	log plain
-rw-r--r--	generate_proposals_op_util_boxes_test.cc	5336	log plain
-rw-r--r--	generate_proposals_op_util_nms.h	24758	log plain
-rw-r--r--	generate_proposals_op_util_nms_gpu.cu	18595	log plain
-rw-r--r--	generate_proposals_op_util_nms_gpu.h	2067	log plain
-rw-r--r--	generate_proposals_op_util_nms_gpu_test.cc	21584	log plain
-rw-r--r--	generate_proposals_op_util_nms_test.cc	13377	log plain
-rw-r--r--	given_tensor_byte_string_to_uint8_fill_op.cc	2299	log plain
-rw-r--r--	given_tensor_byte_string_to_uint8_fill_op.cu	248	log plain
-rw-r--r--	given_tensor_byte_string_to_uint8_fill_op.h	2059	log plain
-rw-r--r--	given_tensor_fill_op.cc	7402	log plain
-rw-r--r--	given_tensor_fill_op.cu	556	log plain
-rw-r--r--	given_tensor_fill_op.h	2914	log plain
-rw-r--r--	glu_op.cc	1474	log plain
-rw-r--r--	glu_op.cu	1142	log plain
-rw-r--r--	glu_op.h	1443	log plain
-rw-r--r--	group_norm_op.cc	6765	log plain
-rw-r--r--	group_norm_op.cu	15141	log plain
-rw-r--r--	group_norm_op.h	8298	log plain
-rw-r--r--	gru_unit_op.cc	1894	log plain
-rw-r--r--	gru_unit_op.h	6644	log plain
-rw-r--r--	gru_unit_op_gpu.cu	3572	log plain
-rw-r--r--	h_softmax_op.cc	23217	log plain
-rw-r--r--	h_softmax_op.h	4928	log plain
-rw-r--r--	half_float_ops.cc	1858	log plain
-rw-r--r--	half_float_ops.cu	1413	log plain
-rw-r--r--	half_float_ops.h	2482	log plain
-rw-r--r--	hard_sigmoid_op.cc	3852	log plain
-rw-r--r--	hard_sigmoid_op.cu	2181	log plain
-rw-r--r--	hard_sigmoid_op.h	994	log plain
-rw-r--r--	heatmap_max_keypoint_op.cc	6048	log plain
-rw-r--r--	heatmap_max_keypoint_op.h	818	log plain
d---------	hip	515	log plain
-rw-r--r--	if_op.cc	734	log plain
-rw-r--r--	if_op.h	1755	log plain
-rw-r--r--	if_op_gpu.cc	165	log plain
-rw-r--r--	im2col_op.cc	3835	log plain
-rw-r--r--	im2col_op.h	8913	log plain
-rw-r--r--	im2col_op_gpu.cc	245	log plain
-rw-r--r--	index_hash_ops.cc	1152	log plain
-rw-r--r--	index_hash_ops.h	2103	log plain
-rw-r--r--	index_ops.cc	11441	log plain
-rw-r--r--	index_ops.h	3043	log plain
-rw-r--r--	instance_norm_gradient_op.cc	9992	log plain
-rw-r--r--	instance_norm_op.cc	7120	log plain
-rw-r--r--	instance_norm_op.cu	16578	log plain
-rw-r--r--	instance_norm_op.h	2676	log plain
-rw-r--r--	integral_image_op.cc	5031	log plain
-rw-r--r--	integral_image_op.cu	6808	log plain
-rw-r--r--	integral_image_op.h	875	log plain
-rw-r--r--	is_empty_op.cc	1557	log plain
-rw-r--r--	is_empty_op.h	558	log plain
-rw-r--r--	jsd_op.cc	2848	log plain
-rw-r--r--	jsd_op.h	721	log plain
-rw-r--r--	key_split_ops.cc	308	log plain
-rw-r--r--	key_split_ops.h	1385	log plain
-rw-r--r--	last_n_window_collector.cc	6018	log plain
-rw-r--r--	layer_norm_op.cc	6382	log plain
-rw-r--r--	layer_norm_op.cu	6748	log plain
-rw-r--r--	layer_norm_op.h	6233	log plain
-rw-r--r--	leaky_relu_op.cc	3403	log plain
-rw-r--r--	leaky_relu_op.cu	1667	log plain
-rw-r--r--	leaky_relu_op.h	1079	log plain
-rw-r--r--	length_split_op.cc	1194	log plain
-rw-r--r--	length_split_op.h	2244	log plain
-rw-r--r--	lengths_pad_op.cc	1861	log plain
-rw-r--r--	lengths_pad_op.cu	188	log plain
-rw-r--r--	lengths_pad_op.h	2559	log plain
-rw-r--r--	lengths_reducer_fused_8bit_rowwise_ops.cc	3974	log plain
-rw-r--r--	lengths_reducer_fused_8bit_rowwise_ops.h	2719	log plain
-rw-r--r--	lengths_reducer_ops.cc	5185	log plain
-rw-r--r--	lengths_reducer_ops.h	3508	log plain
-rw-r--r--	lengths_reducer_rowwise_8bit_ops.cc	8053	log plain
-rw-r--r--	lengths_reducer_rowwise_8bit_ops.h	6159	log plain
-rw-r--r--	lengths_tile_op.cc	2821	log plain
-rw-r--r--	lengths_tile_op.cu	3299	log plain
-rw-r--r--	lengths_tile_op.h	582	log plain
-rw-r--r--	lengths_top_k_op.cc	5233	log plain
-rw-r--r--	lengths_top_k_op.h	1294	log plain
-rw-r--r--	listwise_l2r_op.cc	8594	log plain
-rw-r--r--	listwise_l2r_op.h	1500	log plain
-rw-r--r--	load_save_op.cc	8888	log plain
-rw-r--r--	load_save_op.h	19407	log plain
-rw-r--r--	load_save_op_gpu.cc	632	log plain
-rw-r--r--	local_response_normalization_op.cc	15160	log plain
-rw-r--r--	local_response_normalization_op.cu	11528	log plain
-rw-r--r--	local_response_normalization_op.h	2759	log plain
-rw-r--r--	local_response_normalization_op_cudnn.cc	5756	log plain
-rw-r--r--	locally_connected_op.cc	5364	log plain
-rw-r--r--	locally_connected_op.h	3842	log plain
-rw-r--r--	locally_connected_op_gpu.cc	847	log plain
-rw-r--r--	locally_connected_op_impl.h	26495	log plain
-rw-r--r--	locally_connected_op_util.cc	2108	log plain
-rw-r--r--	locally_connected_op_util.h	1334	log plain
-rw-r--r--	log_op.cc	1936	log plain
-rw-r--r--	log_op.h	431	log plain
-rw-r--r--	log_op_gpu.cc	260	log plain
-rw-r--r--	logit_op.cc	2571	log plain
-rw-r--r--	logit_op.cu	1668	log plain
-rw-r--r--	logit_op.h	1021	log plain
-rw-r--r--	loss_op.cc	1844	log plain
-rw-r--r--	loss_op.cu	287	log plain
-rw-r--r--	loss_op.h	1028	log plain
-rw-r--r--	lp_pool_op.cc	10733	log plain
-rw-r--r--	lp_pool_op.cu	10498	log plain
-rw-r--r--	lpnorm_op.cc	5178	log plain
-rw-r--r--	lpnorm_op.h	1213	log plain
-rw-r--r--	lstm_unit_op.cc	1811	log plain
-rw-r--r--	lstm_unit_op.h	8897	log plain
-rw-r--r--	lstm_unit_op_gpu.cu	7147	log plain
-rw-r--r--	map_ops.cc	2494	log plain
-rw-r--r--	map_ops.h	7935	log plain
-rw-r--r--	margin_ranking_criterion_op.cc	3521	log plain
-rw-r--r--	margin_ranking_criterion_op.cu	2633	log plain
-rw-r--r--	margin_ranking_criterion_op.h	1047	log plain
-rw-r--r--	matmul_op.cc	7097	log plain
-rw-r--r--	matmul_op.h	2828	log plain
-rw-r--r--	matmul_op_gpu.cc	164	log plain
-rw-r--r--	max_pool_with_index.cu	7722	log plain
-rw-r--r--	max_pool_with_index_gpu.h	1155	log plain
-rw-r--r--	mean_op.cc	2548	log plain
-rw-r--r--	mean_op.cu	237	log plain
-rw-r--r--	mean_op.h	3021	log plain
-rw-r--r--	mem_query_op.cu	1527	log plain
-rw-r--r--	merge_id_lists_op.cc	1430	log plain
-rw-r--r--	merge_id_lists_op.h	2416	log plain
-rw-r--r--	minmax_gradient_ops.cc	2267	log plain
-rw-r--r--	minmax_ops.cc	4623	log plain
-rw-r--r--	minmax_ops.h	2534	log plain
-rw-r--r--	mod_op.cc	2276	log plain
-rw-r--r--	mod_op.h	969	log plain
-rw-r--r--	moments_op.cc	2600	log plain
-rw-r--r--	moments_op.cu	3080	log plain
-rw-r--r--	moments_op.h	3726	log plain
-rw-r--r--	multi_class_accuracy_op.cc	2295	log plain
-rw-r--r--	multi_class_accuracy_op.cu	2175	log plain
-rw-r--r--	multi_class_accuracy_op.h	539	log plain
-rw-r--r--	negate_gradient_op.cc	802	log plain
-rw-r--r--	negate_gradient_op.h	566	log plain
-rw-r--r--	negate_gradient_op_gpu.cc	198	log plain
-rw-r--r--	negative_op.cc	1709	log plain
-rw-r--r--	negative_op.h	451	log plain
-rw-r--r--	negative_op_gpu.cc	269	log plain
-rw-r--r--	ngram_ops.cc	357	log plain
-rw-r--r--	ngram_ops.h	2629	log plain
-rw-r--r--	no_default_engine_op.h	1063	log plain
-rw-r--r--	norm_planar_yuv_op.cc	1566	log plain
-rw-r--r--	normalize_l1_op.cc	1156	log plain
-rw-r--r--	normalize_l1_op.h	1075	log plain
-rw-r--r--	normalize_op.cc	2778	log plain
-rw-r--r--	normalize_op.h	2283	log plain
-rw-r--r--	normalize_ops.cu	4972	log plain
-rw-r--r--	numpy_tile_op.cc	503	log plain
-rw-r--r--	numpy_tile_op.h	3576	log plain
-rw-r--r--	one_hot_ops.cc	11896	log plain
-rw-r--r--	one_hot_ops.cu	929	log plain
-rw-r--r--	one_hot_ops.h	2411	log plain
-rw-r--r--	onnx_while_op.cc	3351	log plain
-rw-r--r--	onnx_while_op.h	10540	log plain
-rw-r--r--	onnxifi_op.cc	8007	log plain
-rw-r--r--	onnxifi_op.h	8534	log plain
-rw-r--r--	op_utils_cudnn.h	2112	log plain
-rw-r--r--	operator_fallback_gpu.h	4146	log plain
-rw-r--r--	operator_fallback_gpu_test.cc	2538	log plain
-rw-r--r--	order_switch_ops.cc	3120	log plain
-rw-r--r--	order_switch_ops.h	2149	log plain
-rw-r--r--	order_switch_ops_cudnn.cc	5817	log plain
-rw-r--r--	order_switch_ops_gpu.cc	265	log plain
-rw-r--r--	pack_rnn_sequence_op.cc	3476	log plain
-rw-r--r--	pack_rnn_sequence_op.h	3059	log plain
-rw-r--r--	pack_segments.cc	6981	log plain
-rw-r--r--	pack_segments.cu	9818	log plain
-rw-r--r--	pack_segments.h	2547	log plain
-rw-r--r--	pad_op.cc	15401	log plain
-rw-r--r--	pad_op.h	2873	log plain
-rw-r--r--	pad_op_gpu.cu	16566	log plain
-rw-r--r--	partition_ops.cc	4171	log plain
-rw-r--r--	partition_ops.h	9898	log plain
-rw-r--r--	percentile_op.cc	5949	log plain
-rw-r--r--	percentile_op.h	895	log plain
-rw-r--r--	perplexity_op.cc	1259	log plain
-rw-r--r--	perplexity_op.cu	1190	log plain
-rw-r--r--	perplexity_op.h	447	log plain
-rw-r--r--	piecewise_linear_transform_op.cc	3491	log plain
-rw-r--r--	piecewise_linear_transform_op.cu	8023	log plain
-rw-r--r--	piecewise_linear_transform_op.h	8153	log plain
-rw-r--r--	pool_gradient_op.cc	26669	log plain
-rw-r--r--	pool_op.cc	42490	log plain
-rw-r--r--	pool_op.cu	60182	log plain
-rw-r--r--	pool_op.h	8495	log plain
-rw-r--r--	pool_op_cudnn.cc	17711	log plain
-rw-r--r--	pool_op_util.cc	11285	log plain
-rw-r--r--	pool_op_util.h	1105	log plain
-rw-r--r--	pow_op.cc	12444	log plain
-rw-r--r--	pow_op.cu	2660	log plain
-rw-r--r--	pow_op.h	4664	log plain
-rw-r--r--	prefetch_op.h	4655	log plain
-rw-r--r--	prelu_op.cc	10257	log plain
-rw-r--r--	prelu_op.cu	7428	log plain
-rw-r--r--	prelu_op.h	1037	log plain
-rw-r--r--	prepend_dim_op.cc	1308	log plain
-rw-r--r--	prepend_dim_op.h	2624	log plain
-rw-r--r--	prepend_dim_op_gpu.cc	248	log plain
-rw-r--r--	quant_decode_op.cc	2147	log plain
-rw-r--r--	quant_decode_op.h	5340	log plain
d---------	quantized	2228	log plain
-rw-r--r--	rank_loss_op.cc	6217	log plain
-rw-r--r--	rank_loss_op.h	820	log plain
-rw-r--r--	reciprocal_gradient_op.cc	1311	log plain
-rw-r--r--	reciprocal_op.cc	1345	log plain
-rw-r--r--	reciprocal_op.cu	1341	log plain
-rw-r--r--	reciprocal_op.h	721	log plain
-rw-r--r--	reduce_front_back_max_ops.cc	8403	log plain
-rw-r--r--	reduce_front_back_max_ops.cu	4715	log plain
-rw-r--r--	reduce_front_back_max_ops.h	4377	log plain
-rw-r--r--	reduce_front_back_mean_ops.cc	8526	log plain
-rw-r--r--	reduce_front_back_sum_mean_ops.cu	7047	log plain
-rw-r--r--	reduce_front_back_sum_mean_ops.h	5315	log plain
-rw-r--r--	reduce_front_back_sum_ops.cc	8300	log plain
-rw-r--r--	reduce_ops.cc	14010	log plain
-rw-r--r--	reduce_ops.cu	4508	log plain
-rw-r--r--	reduce_ops.h	8457	log plain
-rw-r--r--	reducer_functors.h	24551	log plain
-rw-r--r--	reduction_ops.cc	9524	log plain
-rw-r--r--	reduction_ops.cu	3736	log plain
-rw-r--r--	reduction_ops.h	4720	log plain
-rw-r--r--	relu_n_op.cc	2785	log plain
-rw-r--r--	relu_n_op.cu	2057	log plain
-rw-r--r--	relu_n_op.h	990	log plain
-rw-r--r--	relu_op.cc	3958	log plain
-rw-r--r--	relu_op.cu	5069	log plain
-rw-r--r--	relu_op.h	624	log plain
-rw-r--r--	relu_op_cudnn.cc	305	log plain
-rw-r--r--	remove_data_blocks_op.cc	798	log plain
-rw-r--r--	remove_data_blocks_op.h	2651	log plain
-rw-r--r--	replace_nan_op.cc	854	log plain
-rw-r--r--	replace_nan_op.cu	745	log plain
-rw-r--r--	replace_nan_op.h	1155	log plain
-rw-r--r--	reservoir_sampling.cc	9048	log plain
-rw-r--r--	reshape_op.cc	5796	log plain
-rw-r--r--	reshape_op.h	4227	log plain
-rw-r--r--	reshape_op_gpu.cc	217	log plain
-rw-r--r--	reshape_op_gpu_test.cc	1457	log plain
-rw-r--r--	resize_op.cc	11195	log plain
-rw-r--r--	resize_op.cu	4759	log plain
-rw-r--r--	resize_op.h	2176	log plain
-rw-r--r--	reverse_packed_segs_op.cc	1119	log plain
-rw-r--r--	reverse_packed_segs_op.cu	2530	log plain
-rw-r--r--	reverse_packed_segs_op.h	2772	log plain
-rw-r--r--	rmac_regions_op.cc	3643	log plain
-rw-r--r--	rmac_regions_op.cu	5872	log plain
-rw-r--r--	rmac_regions_op.h	693	log plain
d---------	rnn	815	log plain
-rw-r--r--	roi_align_gradient_op.cc	7602	log plain
-rw-r--r--	roi_align_gradient_op.cu	7021	log plain
-rw-r--r--	roi_align_gradient_op.h	1178	log plain
-rw-r--r--	roi_align_op.cc	11677	log plain
-rw-r--r--	roi_align_op.cu	5395	log plain
-rw-r--r--	roi_align_op.h	1387	log plain
-rw-r--r--	roi_align_op_gpu_test.cc	8972	log plain
-rw-r--r--	roi_align_rotated_gradient_op.cc	924	log plain
-rw-r--r--	roi_align_rotated_gradient_op.cu	6969	log plain
-rw-r--r--	roi_align_rotated_gradient_op.h	1243	log plain
-rw-r--r--	roi_align_rotated_op.cc	12286	log plain
-rw-r--r--	roi_align_rotated_op.cu	5681	log plain
-rw-r--r--	roi_align_rotated_op.h	1425	log plain
-rw-r--r--	roi_pool_op.cc	7358	log plain
-rw-r--r--	roi_pool_op.cu	6904	log plain
-rw-r--r--	roi_pool_op.h	2442	log plain
-rw-r--r--	rowmul_op.cc	1550	log plain
-rw-r--r--	rowmul_op.h	1947	log plain
-rw-r--r--	rsqrt_op.cc	1758	log plain
-rw-r--r--	rsqrt_op.cu	1416	log plain
-rw-r--r--	rsqrt_op.h	729	log plain
-rw-r--r--	scale_op.cc	852	log plain
-rw-r--r--	scale_op.h	996	log plain
-rw-r--r--	scale_op_gpu.cc	309	log plain
-rw-r--r--	segment_reduction_op.cc	20662	log plain
-rw-r--r--	segment_reduction_op.h	70493	log plain
-rw-r--r--	segment_reduction_op_gpu.cu	57134	log plain
-rw-r--r--	selu_op.cc	4131	log plain
-rw-r--r--	selu_op.cu	1884	log plain
-rw-r--r--	selu_op.h	1515	log plain
-rw-r--r--	sequence_ops.cc	15958	log plain
-rw-r--r--	sequence_ops.cu	10848	log plain
-rw-r--r--	sequence_ops.h	8204	log plain
-rw-r--r--	shape_op.cc	1769	log plain
-rw-r--r--	shape_op.h	1623	log plain
-rw-r--r--	shape_op_gpu.cc	151	log plain
-rw-r--r--	sigmoid_gradient_op.cc	1297	log plain
-rw-r--r--	sigmoid_op.cc	1960	log plain
-rw-r--r--	sigmoid_op.cu	1977	log plain
-rw-r--r--	sigmoid_op.h	639	log plain
-rw-r--r--	sigmoid_op_cudnn.cc	320	log plain
-rw-r--r--	sin_op.cc	2346	log plain
-rw-r--r--	sin_op.cu	1267	log plain
-rw-r--r--	sin_op.h	705	log plain
-rw-r--r--	sinh_op.cc	2406	log plain
-rw-r--r--	sinh_op.cu	1274	log plain
-rw-r--r--	sinh_op.h	711	log plain
-rw-r--r--	sinusoid_position_encoding_op.cc	1404	log plain
-rw-r--r--	sinusoid_position_encoding_op.h	2811	log plain
-rw-r--r--	slice_op.cc	3720	log plain
-rw-r--r--	slice_op.cu	10822	log plain
-rw-r--r--	slice_op.h	9974	log plain
-rw-r--r--	softmax_op.cc	6327	log plain
-rw-r--r--	softmax_op.h	1141	log plain
-rw-r--r--	softmax_op_cudnn.cc	3966	log plain
-rw-r--r--	softmax_ops.cu	23598	log plain
-rw-r--r--	softmax_shared.cc	1375	log plain
-rw-r--r--	softmax_shared.h	462	log plain
-rw-r--r--	softmax_with_loss_op.cc	12895	log plain
-rw-r--r--	softmax_with_loss_op.h	2525	log plain
-rw-r--r--	softplus_op.cc	2977	log plain
-rw-r--r--	softplus_op.cu	1609	log plain
-rw-r--r--	softplus_op.h	781	log plain
-rw-r--r--	softsign_op.cc	3623	log plain
-rw-r--r--	softsign_op.cu	2031	log plain
-rw-r--r--	softsign_op.h	675	log plain
-rw-r--r--	space_batch_op.cc	3909	log plain
-rw-r--r--	space_batch_op.h	6833	log plain
-rw-r--r--	space_batch_op_gpu.cu	4852	log plain
-rw-r--r--	sparse_normalize_op.cc	2535	log plain
-rw-r--r--	sparse_normalize_op.h	826	log plain
-rw-r--r--	sparse_normalize_op_gpu.cu	226	log plain
-rw-r--r--	sparse_to_dense_mask_op.cc	4261	log plain
-rw-r--r--	sparse_to_dense_mask_op.h	9592	log plain
-rw-r--r--	sparse_to_dense_op.cc	1914	log plain
-rw-r--r--	sparse_to_dense_op.cu	2387	log plain
-rw-r--r--	sparse_to_dense_op.h	3962	log plain
-rw-r--r--	spatial_batch_norm_gradient_op.cc	6326	log plain
-rw-r--r--	spatial_batch_norm_op.cc	6003	log plain
-rw-r--r--	spatial_batch_norm_op.cu	294	log plain
-rw-r--r--	spatial_batch_norm_op.h	14850	log plain
-rw-r--r--	spatial_batch_norm_op_cudnn.cu	12449	log plain
-rw-r--r--	spatial_batch_norm_op_impl.cuh	16338	log plain
-rw-r--r--	spatial_softmax_with_loss_op.cc	8373	log plain
-rw-r--r--	spatial_softmax_with_loss_op.h	2134	log plain
-rw-r--r--	sqr_op.cc	2107	log plain
-rw-r--r--	sqr_op.h	431	log plain
-rw-r--r--	sqr_op_gpu.cc	260	log plain
-rw-r--r--	sqrt_op.cc	2226	log plain
-rw-r--r--	sqrt_op.h	448	log plain
-rw-r--r--	sqrt_op_gpu.cc	263	log plain
-rw-r--r--	square_root_divide_op.cc	1083	log plain
-rw-r--r--	square_root_divide_op.h	1788	log plain
-rw-r--r--	stats_ops.cc	9858	log plain
-rw-r--r--	stats_put_ops.cc	3630	log plain
-rw-r--r--	stats_put_ops.h	2748	log plain
-rw-r--r--	stop_gradient.cc	551	log plain
-rw-r--r--	stop_gradient.h	548	log plain
-rw-r--r--	stop_gradient_gpu.cc	191	log plain
-rw-r--r--	string_ops.cc	6210	log plain
-rw-r--r--	string_ops.h	2051	log plain
-rw-r--r--	string_ops_test.cc	4087	log plain
-rw-r--r--	stump_func_op.cc	2724	log plain
-rw-r--r--	stump_func_op.cu	1589	log plain
-rw-r--r--	stump_func_op.h	2082	log plain
-rw-r--r--	stylizer_ops.cc	19473	log plain
-rw-r--r--	summarize_op.cc	2285	log plain
-rw-r--r--	summarize_op.cu	3385	log plain
-rw-r--r--	summarize_op.h	1866	log plain
-rw-r--r--	swish_op.cc	2662	log plain
-rw-r--r--	swish_op.cu	2128	log plain
-rw-r--r--	swish_op.h	772	log plain
-rw-r--r--	tan_op.cc	1742	log plain
-rw-r--r--	tan_op.cu	1345	log plain
-rw-r--r--	tan_op.h	705	log plain
-rw-r--r--	tanh_gradient_op.cc	1286	log plain
-rw-r--r--	tanh_op.cc	1848	log plain
-rw-r--r--	tanh_op.cu	1305	log plain
-rw-r--r--	tanh_op.h	723	log plain
-rw-r--r--	tanh_op_cudnn.cc	305	log plain
-rw-r--r--	tensor_protos_db_input.cc	1701	log plain
-rw-r--r--	tensor_protos_db_input.h	3633	log plain
-rw-r--r--	tensor_protos_db_input_gpu.cc	248	log plain
-rw-r--r--	text_file_reader.cc	6351	log plain
-rw-r--r--	text_file_reader_utils.cc	3093	log plain
-rw-r--r--	text_file_reader_utils.h	2907	log plain
-rw-r--r--	text_file_reader_utils_test.cc	4243	log plain
-rw-r--r--	thresholded_relu_op.cc	2902	log plain
-rw-r--r--	thresholded_relu_op.cu	1642	log plain
-rw-r--r--	thresholded_relu_op.h	1107	log plain
-rw-r--r--	tile_op.cc	3423	log plain
-rw-r--r--	tile_op.cu	2507	log plain
-rw-r--r--	tile_op.h	8143	log plain
-rw-r--r--	top_k.cc	9924	log plain
-rw-r--r--	top_k.cu	10794	log plain
-rw-r--r--	top_k.h	1086	log plain
-rw-r--r--	top_k_heap_selection.cuh	8923	log plain
-rw-r--r--	top_k_radix_selection.cuh	15557	log plain
-rw-r--r--	transpose_op.cc	3434	log plain
-rw-r--r--	transpose_op.cu	185	log plain
-rw-r--r--	transpose_op.h	1977	log plain
-rw-r--r--	transpose_op_cudnn.cc	4259	log plain
-rw-r--r--	tt_linear_op.cc	2569	log plain
-rw-r--r--	tt_linear_op.h	6418	log plain
-rw-r--r--	unique_ops.cc	3534	log plain
-rw-r--r--	unique_ops.cu	4252	log plain
-rw-r--r--	unique_ops.h	1666	log plain
-rw-r--r--	upsample_op.cc	6978	log plain
-rw-r--r--	upsample_op.cu	7926	log plain
-rw-r--r--	upsample_op.h	2184	log plain
-rw-r--r--	utility_ops.cc	30832	log plain
-rw-r--r--	utility_ops.cu	11817	log plain
-rw-r--r--	utility_ops.h	41918	log plain
-rw-r--r--	utility_ops_cudnn.cc	5424	log plain
-rw-r--r--	utility_ops_gpu_test.cc	1427	log plain
-rw-r--r--	utility_ops_test.cc	1223	log plain
-rw-r--r--	variable_length_sequence_padding.cc	713	log plain
-rw-r--r--	variable_length_sequence_padding.h	1376	log plain
-rw-r--r--	weighted_multi_sampling_op.cc	4034	log plain
-rw-r--r--	weighted_multi_sampling_op.h	587	log plain
-rw-r--r--	weighted_sample_op.cc	4109	log plain
-rw-r--r--	weighted_sample_op.cu	2842	log plain
-rw-r--r--	weighted_sample_op.h	724	log plain
-rw-r--r--	while_op.cc	841	log plain
-rw-r--r--	while_op.h	1952	log plain
-rw-r--r--	while_op_gpu.cc	174	log plain
-rw-r--r--	workspace_ops.cc	1258	log plain
-rw-r--r--	zero_gradient_op.cc	797	log plain
-rw-r--r--	zero_gradient_op.h	347	log plain
-rw-r--r--	zero_gradient_op_gpu.cc	173	log plain