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

Mode	Name	Size
-rw-r--r--	CMakeLists.txt	1464	log plain
-rw-r--r--	abs_op.cc	1714	log plain
-rw-r--r--	abs_op.cu	1439	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	1200	log plain
-rw-r--r--	accuracy_op.cc	2409	log plain
-rw-r--r--	accuracy_op.cu	2115	log plain
-rw-r--r--	accuracy_op.h	645	log plain
-rw-r--r--	apmeter_op.cc	3600	log plain
-rw-r--r--	apmeter_op.h	1011	log plain
-rw-r--r--	arg_max_op.cc	1272	log plain
-rw-r--r--	arg_max_op.h	550	log plain
-rw-r--r--	arg_ops.cc	3760	log plain
-rw-r--r--	arg_ops.cu	3293	log plain
-rw-r--r--	arg_ops.h	2560	log plain
-rw-r--r--	arg_ops_eigen.h	2424	log plain
-rw-r--r--	assert_op.cc	557	log plain
-rw-r--r--	assert_op.cu	176	log plain
-rw-r--r--	assert_op.h	1315	log plain
-rw-r--r--	atomic_ops.cc	4449	log plain
-rw-r--r--	batch_box_cox_op.cc	11219	log plain
-rw-r--r--	batch_box_cox_op.h	2167	log plain
-rw-r--r--	batch_gather_ops.cc	2150	log plain
-rw-r--r--	batch_gather_ops.cu	5200	log plain
-rw-r--r--	batch_gather_ops.h	4586	log plain
-rw-r--r--	batch_matmul_op.cc	8845	log plain
-rw-r--r--	batch_matmul_op.cu	664	log plain
-rw-r--r--	batch_matmul_op.h	9357	log plain
-rw-r--r--	batch_matmul_op_gpu_test.cc	2522	log plain
-rw-r--r--	batch_matmul_op_test.cc	2193	log plain
-rw-r--r--	batch_sparse_to_dense_op.cc	7527	log plain
-rw-r--r--	batch_sparse_to_dense_op.h	1261	log plain
-rw-r--r--	bbox_transform_op.cc	5550	log plain
-rw-r--r--	bbox_transform_op.h	1654	log plain
-rw-r--r--	boolean_mask_ops.cc	14359	log plain
-rw-r--r--	boolean_mask_ops.cu	11579	log plain
-rw-r--r--	boolean_mask_ops.h	1975	log plain
-rw-r--r--	boolean_unmask_ops.cc	3092	log plain
-rw-r--r--	boolean_unmask_ops.cu	3301	log plain
-rw-r--r--	boolean_unmask_ops.h	378	log plain
-rw-r--r--	boolean_unmask_ops_test.cc	1629	log plain
-rw-r--r--	box_with_nms_limit_op.cc	9947	log plain
-rw-r--r--	box_with_nms_limit_op.h	2112	log plain
-rw-r--r--	cast_op.cc	5079	log plain
-rw-r--r--	cast_op.cu	3718	log plain
-rw-r--r--	cast_op.h	1384	log plain
-rw-r--r--	ceil_op.cc	632	log plain
-rw-r--r--	ceil_op.cu	713	log plain
-rw-r--r--	ceil_op.h	771	log plain
-rw-r--r--	channel_backprop_stats_op.cc	2921	log plain
-rw-r--r--	channel_backprop_stats_op.cu	5951	log plain
-rw-r--r--	channel_backprop_stats_op.h	740	log plain
-rw-r--r--	channel_shuffle_op.cc	852	log plain
-rw-r--r--	channel_shuffle_op.h	3647	log plain
-rw-r--r--	channel_shuffle_op_gpu.cu	3241	log plain
-rw-r--r--	channel_stats_op.cc	1948	log plain
-rw-r--r--	channel_stats_op.cu	5377	log plain
-rw-r--r--	channel_stats_op.h	714	log plain
-rw-r--r--	clip_op.cc	2296	log plain
-rw-r--r--	clip_op.cu	2139	log plain
-rw-r--r--	clip_op.h	1605	log plain
-rw-r--r--	collect_and_distribute_fpn_rpn_proposals_op.cc	13158	log plain
-rw-r--r--	collect_and_distribute_fpn_rpn_proposals_op.h	3902	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	8026	log plain
-rw-r--r--	concat_split_op.h	8020	log plain
-rw-r--r--	concat_split_op_gpu.cc	388	log plain
-rw-r--r--	conditional_op.cc	2181	log plain
-rw-r--r--	conditional_op.h	462	log plain
-rw-r--r--	conv_gradient_op.cc	2242	log plain
-rw-r--r--	conv_op.cc	3649	log plain
-rw-r--r--	conv_op.h	2499	log plain
-rw-r--r--	conv_op_cache_cudnn.cc	419	log plain
-rw-r--r--	conv_op_cache_cudnn.h	1978	log plain
-rw-r--r--	conv_op_cache_cudnn_test.cc	1560	log plain
-rw-r--r--	conv_op_cudnn.cc	46746	log plain
-rw-r--r--	conv_op_eigen.cc	8310	log plain
-rw-r--r--	conv_op_gpu.cc	704	log plain
-rw-r--r--	conv_op_impl.h	21280	log plain
-rw-r--r--	conv_op_shared.cc	1054	log plain
-rw-r--r--	conv_op_shared.h	690	log plain
-rw-r--r--	conv_op_shared_gpu.cc	906	log plain
-rw-r--r--	conv_pool_op_base.h	26314	log plain
-rw-r--r--	conv_transpose_gradient_op.cc	1648	log plain
-rw-r--r--	conv_transpose_op.cc	2131	log plain
-rw-r--r--	conv_transpose_op.h	1732	log plain
-rw-r--r--	conv_transpose_op_cudnn.cc	22056	log plain
-rw-r--r--	conv_transpose_op_gpu.cc	357	log plain
-rw-r--r--	conv_transpose_op_impl.h	16937	log plain
-rw-r--r--	conv_transpose_op_mobile.cc	434	log plain
-rw-r--r--	conv_transpose_op_mobile.h	1524	log plain
-rw-r--r--	conv_transpose_op_mobile_impl.h	19474	log plain
-rw-r--r--	conv_transpose_op_mobile_test.cc	6788	log plain
-rw-r--r--	conv_transpose_unpool_op_base.h	9567	log plain
-rw-r--r--	cos_op.cc	1618	log plain
-rw-r--r--	cos_op.cu	1404	log plain
-rw-r--r--	cosine_embedding_criterion_op.cc	2531	log plain
-rw-r--r--	cosine_embedding_criterion_op.cu	2038	log plain
-rw-r--r--	cosine_embedding_criterion_op.h	1061	log plain
-rw-r--r--	counter_ops.cc	4902	log plain
-rw-r--r--	counter_ops.h	4613	log plain
-rw-r--r--	counter_ops_gpu.cc	565	log plain
-rw-r--r--	create_scope_op.cc	1240	log plain
-rw-r--r--	create_scope_op.h	5175	log plain
-rw-r--r--	cross_entropy_op.cc	19189	log plain
-rw-r--r--	cross_entropy_op.cu	11945	log plain
-rw-r--r--	cross_entropy_op.h	4015	log plain
-rw-r--r--	dataset_ops.cc	48692	log plain
-rw-r--r--	dataset_ops.h	5335	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	21568	log plain
-rw-r--r--	deform_conv_op.h	3525	log plain
-rw-r--r--	deform_conv_op_impl.h	12551	log plain
-rw-r--r--	distance_op.cc	16793	log plain
-rw-r--r--	distance_op.cu	14513	log plain
-rw-r--r--	distance_op.h	8042	log plain
-rw-r--r--	do_op.cc	1193	log plain
-rw-r--r--	do_op.h	6914	log plain
-rw-r--r--	do_op_gpu.cc	165	log plain
-rw-r--r--	dropout_op.cc	3879	log plain
-rw-r--r--	dropout_op.cu	2589	log plain
-rw-r--r--	dropout_op.h	1482	log plain
-rw-r--r--	dropout_op_cudnn.cc	8558	log plain
-rw-r--r--	elementwise_add_op.cc	287	log plain
-rw-r--r--	elementwise_div_op.cc	853	log plain
-rw-r--r--	elementwise_linear_op.cc	3440	log plain
-rw-r--r--	elementwise_linear_op.cu	3335	log plain
-rw-r--r--	elementwise_linear_op.h	1139	log plain
-rw-r--r--	elementwise_logical_ops.cc	2200	log plain
-rw-r--r--	elementwise_logical_ops.h	5043	log plain
-rw-r--r--	elementwise_mul_op.cc	287	log plain
-rw-r--r--	elementwise_op.cc	6911	log plain
-rw-r--r--	elementwise_op.cu	13230	log plain
-rw-r--r--	elementwise_op.h	14682	log plain
-rw-r--r--	elementwise_op_gpu_test.cc	960	log plain
-rw-r--r--	elementwise_op_schema.cc	13043	log plain
-rw-r--r--	elementwise_op_test.cc	767	log plain
-rw-r--r--	elementwise_op_test.h	9203	log plain
-rw-r--r--	elementwise_sub_op.cc	287	log plain
-rw-r--r--	elementwise_sum_op.cc	1050	log plain
-rw-r--r--	elu_op.cc	2455	log plain
-rw-r--r--	elu_op.cu	1751	log plain
-rw-r--r--	elu_op.h	895	log plain
-rw-r--r--	enforce_finite_op.cc	524	log plain
-rw-r--r--	enforce_finite_op.cu	398	log plain
-rw-r--r--	enforce_finite_op.h	1137	log plain
-rw-r--r--	exp_op.cc	1255	log plain
-rw-r--r--	exp_op.cu	754	log plain
-rw-r--r--	expand_squeeze_dims_op.cc	4158	log plain
-rw-r--r--	expand_squeeze_dims_op.h	3333	log plain
-rw-r--r--	expand_squeeze_dims_op_gpu.cc	252	log plain
-rw-r--r--	extend_tensor_op.cc	2477	log plain
-rw-r--r--	fc_inference.cc	1803	log plain
-rw-r--r--	fc_inference.h	429	log plain
-rw-r--r--	feed_blob_op.cc	508	log plain
-rw-r--r--	feed_blob_op.h	731	log plain
-rw-r--r--	filler_op.cc	10083	log plain
-rw-r--r--	filler_op.cu	2042	log plain
-rw-r--r--	filler_op.h	17165	log plain
-rw-r--r--	filler_op_gpu.cc	249	log plain
-rw-r--r--	find_duplicate_elements_op.cc	760	log plain
-rw-r--r--	find_duplicate_elements_op.h	1529	log plain
-rw-r--r--	find_op.cc	739	log plain
-rw-r--r--	find_op.cu	1471	log plain
-rw-r--r--	find_op.h	2039	log plain
-rw-r--r--	flatten_op.cc	1903	log plain
-rw-r--r--	flatten_op.h	969	log plain
-rw-r--r--	flexible_top_k.cc	5740	log plain
-rw-r--r--	flexible_top_k.h	906	log plain
-rw-r--r--	floor_op.cc	640	log plain
-rw-r--r--	floor_op.cu	721	log plain
-rw-r--r--	floor_op.h	777	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	8716	log plain
-rw-r--r--	fully_connected_op.h	8415	log plain
-rw-r--r--	fully_connected_op_gpu.cc	6056	log plain
-rw-r--r--	fused_rowwise_8bit_conversion_ops.cc	2064	log plain
-rw-r--r--	fused_rowwise_8bit_conversion_ops.h	4847	log plain
-rw-r--r--	gather_fused_8bit_rowwise_op.cc	1272	log plain
-rw-r--r--	gather_fused_8bit_rowwise_op.h	2132	log plain
-rw-r--r--	gather_ranges_to_dense_op.cc	2913	log plain
-rw-r--r--	gather_ranges_to_dense_op.h	4459	log plain
-rw-r--r--	generate_proposals_op.cc	11130	log plain
-rw-r--r--	generate_proposals_op.h	3845	log plain
-rw-r--r--	generate_proposals_op_test.cc	10949	log plain
-rw-r--r--	generate_proposals_op_util_boxes.h	4993	log plain
-rw-r--r--	generate_proposals_op_util_boxes_test.cc	1443	log plain
-rw-r--r--	generate_proposals_op_util_nms.h	7707	log plain
-rw-r--r--	generate_proposals_op_util_nms_test.cc	5321	log plain
-rw-r--r--	given_tensor_fill_op.cc	5788	log plain
-rw-r--r--	given_tensor_fill_op.cu	556	log plain
-rw-r--r--	given_tensor_fill_op.h	2877	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	1437	log plain
-rw-r--r--	group_norm_op.cc	2101	log plain
-rw-r--r--	group_norm_op.cu	16496	log plain
-rw-r--r--	group_norm_op.h	1604	log plain
-rw-r--r--	gru_unit_op.cc	1894	log plain
-rw-r--r--	gru_unit_op.h	6600	log plain
-rw-r--r--	gru_unit_op_gpu.cu	3555	log plain
-rw-r--r--	h_softmax_op.cc	21961	log plain
-rw-r--r--	h_softmax_op.h	4874	log plain
-rw-r--r--	half_float_ops.cc	1858	log plain
-rw-r--r--	half_float_ops.cu	1362	log plain
-rw-r--r--	half_float_ops.h	1508	log plain
-rw-r--r--	if_op.cc	734	log plain
-rw-r--r--	if_op.h	1748	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	8796	log plain
-rw-r--r--	im2col_op_gpu.cc	245	log plain
-rw-r--r--	index_hash_ops.cc	1151	log plain
-rw-r--r--	index_hash_ops.h	2099	log plain
-rw-r--r--	index_ops.cc	13629	log plain
-rw-r--r--	instance_norm_gradient_op.cc	9430	log plain
-rw-r--r--	instance_norm_op.cc	5582	log plain
-rw-r--r--	instance_norm_op.cu	16364	log plain
-rw-r--r--	instance_norm_op.h	2656	log plain
-rw-r--r--	jsd_op.cc	2805	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	1382	log plain
-rw-r--r--	last_n_window_collector.cc	5835	log plain
-rw-r--r--	layer_norm_op.cc	7719	log plain
-rw-r--r--	layer_norm_op.cu	9885	log plain
-rw-r--r--	layer_norm_op.h	1606	log plain
-rw-r--r--	leaky_relu_op.cc	2359	log plain
-rw-r--r--	leaky_relu_op.cu	1619	log plain
-rw-r--r--	leaky_relu_op.h	1077	log plain
-rw-r--r--	lengths_reducer_fused_8bit_rowwise_ops.cc	3171	log plain
-rw-r--r--	lengths_reducer_fused_8bit_rowwise_ops.h	2722	log plain
-rw-r--r--	lengths_reducer_ops.cc	1919	log plain
-rw-r--r--	lengths_reducer_ops.h	3510	log plain
-rw-r--r--	lengths_reducer_rowwise_8bit_ops.cc	7168	log plain
-rw-r--r--	lengths_reducer_rowwise_8bit_ops.h	6095	log plain
-rw-r--r--	lengths_tile_op.cc	1467	log plain
-rw-r--r--	lengths_tile_op.h	1827	log plain
-rw-r--r--	lengths_tile_op_gpu.cc	190	log plain
-rw-r--r--	lengths_top_k_op.cc	5428	log plain
-rw-r--r--	lengths_top_k_op.h	1294	log plain
-rw-r--r--	listwise_l2r_op.cc	6665	log plain
-rw-r--r--	listwise_l2r_op.h	1132	log plain
-rw-r--r--	load_save_op.cc	5825	log plain
-rw-r--r--	load_save_op.h	19189	log plain
-rw-r--r--	load_save_op_gpu.cc	576	log plain
-rw-r--r--	local_response_normalization_op.cc	11372	log plain
-rw-r--r--	local_response_normalization_op.cu	11321	log plain
-rw-r--r--	local_response_normalization_op.h	2738	log plain
-rw-r--r--	local_response_normalization_op_cudnn.cc	5734	log plain
-rw-r--r--	locally_connected_op.cc	5300	log plain
-rw-r--r--	locally_connected_op.h	4000	log plain
-rw-r--r--	locally_connected_op_gpu.cc	847	log plain
-rw-r--r--	locally_connected_op_impl.h	26088	log plain
-rw-r--r--	locally_connected_op_util.cc	1548	log plain
-rw-r--r--	locally_connected_op_util.h	1196	log plain
-rw-r--r--	log_op.cc	1251	log plain
-rw-r--r--	log_op.cu	407	log plain
-rw-r--r--	logit_op.cc	2560	log plain
-rw-r--r--	logit_op.cu	1792	log plain
-rw-r--r--	logit_op.h	546	log plain
-rw-r--r--	loss_op.cc	1107	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	9331	log plain
-rw-r--r--	lp_pool_op.cu	10555	log plain
-rw-r--r--	lpnorm_op.cc	3661	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	8951	log plain
-rw-r--r--	lstm_unit_op_gpu.cu	7110	log plain
-rw-r--r--	map_ops.cc	2446	log plain
-rw-r--r--	map_ops.h	7829	log plain
-rw-r--r--	margin_ranking_criterion_op.cc	3420	log plain
-rw-r--r--	margin_ranking_criterion_op.cu	2569	log plain
-rw-r--r--	margin_ranking_criterion_op.h	1047	log plain
-rw-r--r--	math_ops.cc	2087	log plain
-rw-r--r--	math_ops.cu	962	log plain
-rw-r--r--	math_ops.h	325	log plain
-rw-r--r--	matmul_op.cc	5855	log plain
-rw-r--r--	matmul_op.h	2823	log plain
-rw-r--r--	matmul_op_gpu.cc	164	log plain
-rw-r--r--	max_pool_with_index.cu	7612	log plain
-rw-r--r--	max_pool_with_index.h	1289	log plain
-rw-r--r--	mean_op.cc	1291	log plain
-rw-r--r--	mean_op.cu	237	log plain
-rw-r--r--	mean_op.h	2996	log plain
-rw-r--r--	mem_query_op.cu	1504	log plain
-rw-r--r--	merge_id_lists_op.cc	1430	log plain
-rw-r--r--	merge_id_lists_op.h	2421	log plain
-rw-r--r--	minmax_gradient_ops.cc	2229	log plain
-rw-r--r--	minmax_ops.cc	2239	log plain
-rw-r--r--	minmax_ops.h	2523	log plain
-rw-r--r--	mod_op.cc	1472	log plain
-rw-r--r--	mod_op.h	967	log plain
-rw-r--r--	multi_class_accuracy_op.cc	2264	log plain
-rw-r--r--	multi_class_accuracy_op.cu	2139	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	559	log plain
-rw-r--r--	negate_gradient_op_gpu.cc	198	log plain
-rw-r--r--	negative_op.cc	1161	log plain
-rw-r--r--	negative_op.cu	736	log plain
-rw-r--r--	ngram_ops.cc	357	log plain
-rw-r--r--	ngram_ops.h	2625	log plain
-rw-r--r--	no_default_engine_op.h	1063	log plain
-rw-r--r--	norm_planar_yuv_op.cc	1503	log plain
-rw-r--r--	normalize_l1_op.cc	1118	log plain
-rw-r--r--	normalize_l1_op.h	1067	log plain
-rw-r--r--	normalize_op.cc	2712	log plain
-rw-r--r--	normalize_op.h	2200	log plain
-rw-r--r--	normalize_ops.cu	4826	log plain
-rw-r--r--	one_hot_ops.cc	9617	log plain
-rw-r--r--	one_hot_ops.cu	939	log plain
-rw-r--r--	one_hot_ops.h	2426	log plain
-rw-r--r--	onnx_while_op.cc	2354	log plain
-rw-r--r--	onnx_while_op.h	10148	log plain
-rw-r--r--	op_utils_cudnn.h	2112	log plain
-rw-r--r--	operator_fallback_gpu.h	4238	log plain
-rw-r--r--	operator_fallback_gpu_test.cc	2550	log plain
-rw-r--r--	order_switch_ops.cc	3251	log plain
-rw-r--r--	order_switch_ops.cu	1750	log plain
-rw-r--r--	order_switch_ops.h	851	log plain
-rw-r--r--	pack_rnn_sequence_op.cc	3476	log plain
-rw-r--r--	pack_rnn_sequence_op.h	3072	log plain
-rw-r--r--	pack_segments.cc	6400	log plain
-rw-r--r--	pack_segments.cu	8621	log plain
-rw-r--r--	pack_segments.h	2213	log plain
-rw-r--r--	pad_op.cc	15276	log plain
-rw-r--r--	pad_op.h	2859	log plain
-rw-r--r--	pad_op_gpu.cu	16514	log plain
-rw-r--r--	partition_ops.cc	4171	log plain
-rw-r--r--	partition_ops.h	9455	log plain
-rw-r--r--	percentile_op.cc	5773	log plain
-rw-r--r--	percentile_op.h	913	log plain
-rw-r--r--	perplexity_op.cc	1243	log plain
-rw-r--r--	perplexity_op.cu	1174	log plain
-rw-r--r--	perplexity_op.h	447	log plain
-rw-r--r--	piecewise_linear_transform_op.cc	3512	log plain
-rw-r--r--	piecewise_linear_transform_op.cu	8026	log plain
-rw-r--r--	piecewise_linear_transform_op.h	8056	log plain
-rw-r--r--	pool_gradient_op.cc	13334	log plain
-rw-r--r--	pool_op.cc	27968	log plain
-rw-r--r--	pool_op.cu	52477	log plain
-rw-r--r--	pool_op.h	1615	log plain
-rw-r--r--	pool_op_cudnn.cu	17061	log plain
-rw-r--r--	pow_op.cc	11373	log plain
-rw-r--r--	pow_op.cu	2497	log plain
-rw-r--r--	pow_op.h	4531	log plain
-rw-r--r--	prefetch_op.h	4658	log plain
-rw-r--r--	prelu_op.cc	8824	log plain
-rw-r--r--	prelu_op.cu	7352	log plain
-rw-r--r--	prelu_op.h	1035	log plain
-rw-r--r--	prepend_dim_op.cc	1308	log plain
-rw-r--r--	prepend_dim_op.h	2598	log plain
-rw-r--r--	prepend_dim_op_gpu.cc	248	log plain
-rw-r--r--	quant_decode_op.cc	2132	log plain
-rw-r--r--	quant_decode_op.h	5333	log plain
-rw-r--r--	rank_loss_op.cc	6181	log plain
-rw-r--r--	rank_loss_op.h	820	log plain
-rw-r--r--	reduce_ops.cc	5528	log plain
-rw-r--r--	reduce_ops.cu	5563	log plain
-rw-r--r--	reduce_ops.h	8221	log plain
-rw-r--r--	reducer_functors.h	24622	log plain
-rw-r--r--	reduction_front_back_ops.cc	15500	log plain
-rw-r--r--	reduction_front_back_ops.cu	11578	log plain
-rw-r--r--	reduction_front_back_ops.h	9138	log plain
-rw-r--r--	reduction_ops.cc	5798	log plain
-rw-r--r--	reduction_ops.cu	3664	log plain
-rw-r--r--	reduction_ops.h	4663	log plain
-rw-r--r--	relu_op.cc	3001	log plain
-rw-r--r--	relu_op.cu	1422	log plain
-rw-r--r--	relu_op.h	755	log plain
-rw-r--r--	relu_op_cudnn.cc	6693	log plain
-rw-r--r--	relu_op_fp16.cu	2765	log plain
-rw-r--r--	remove_data_blocks_op.cc	798	log plain
-rw-r--r--	remove_data_blocks_op.h	2657	log plain
-rw-r--r--	replace_nan_op.cc	852	log plain
-rw-r--r--	replace_nan_op.cu	743	log plain
-rw-r--r--	replace_nan_op.h	1151	log plain
-rw-r--r--	reservoir_sampling.cc	8616	log plain
-rw-r--r--	reshape_op.cc	4787	log plain
-rw-r--r--	reshape_op.h	3937	log plain
-rw-r--r--	reshape_op_gpu.cc	217	log plain
-rw-r--r--	reshape_op_gpu_test.cc	1469	log plain
-rw-r--r--	resize_op.cc	5761	log plain
-rw-r--r--	resize_op.cu	4071	log plain
-rw-r--r--	resize_op.h	1556	log plain
-rw-r--r--	reverse_packed_segs_op.cc	1119	log plain
-rw-r--r--	reverse_packed_segs_op.cu	2553	log plain
-rw-r--r--	reverse_packed_segs_op.h	2807	log plain
-rw-r--r--	rmac_regions_op.cc	3595	log plain
-rw-r--r--	rmac_regions_op.cu	5835	log plain
-rw-r--r--	rmac_regions_op.h	700	log plain
d---------	rnn	844	log plain
-rw-r--r--	roi_align_gradient_op.cc	7550	log plain
-rw-r--r--	roi_align_gradient_op.cu	6906	log plain
-rw-r--r--	roi_align_gradient_op.h	1174	log plain
-rw-r--r--	roi_align_op.cc	11850	log plain
-rw-r--r--	roi_align_op.cu	5237	log plain
-rw-r--r--	roi_align_op.h	1382	log plain
-rw-r--r--	roi_align_op_gpu_test.cc	8781	log plain
-rw-r--r--	roi_pool_op.cc	7339	log plain
-rw-r--r--	roi_pool_op.cu	6655	log plain
-rw-r--r--	roi_pool_op.h	2433	log plain
-rw-r--r--	rowmul_op.cc	1550	log plain
-rw-r--r--	rowmul_op.h	1947	log plain
-rw-r--r--	scale_op.cc	854	log plain
-rw-r--r--	scale_op.h	981	log plain
-rw-r--r--	scale_op_gpu.cc	308	log plain
-rw-r--r--	segment_reduction_op.cc	13936	log plain
-rw-r--r--	segment_reduction_op.h	69624	log plain
-rw-r--r--	segment_reduction_op_gpu.cu	49043	log plain
-rw-r--r--	selu_op.cc	2945	log plain
-rw-r--r--	selu_op.cu	1735	log plain
-rw-r--r--	selu_op.h	1511	log plain
-rw-r--r--	sequence_ops.cc	12969	log plain
-rw-r--r--	sequence_ops.cu	10826	log plain
-rw-r--r--	sequence_ops.h	7769	log plain
-rw-r--r--	shape_op.cc	581	log plain
-rw-r--r--	shape_op.h	818	log plain
-rw-r--r--	shape_op_gpu.cc	151	log plain
-rw-r--r--	sigmoid_op.cc	2020	log plain
-rw-r--r--	sigmoid_op.cu	1458	log plain
-rw-r--r--	sin_op.cc	1588	log plain
-rw-r--r--	sin_op.cu	1403	log plain
-rw-r--r--	sinusoid_position_encoding_op.cc	1404	log plain
-rw-r--r--	sinusoid_position_encoding_op.h	2788	log plain
-rw-r--r--	slice_op.cc	3153	log plain
-rw-r--r--	slice_op.cu	8652	log plain
-rw-r--r--	slice_op.h	9158	log plain
-rw-r--r--	softmax_op.cc	4571	log plain
-rw-r--r--	softmax_op.h	1184	log plain
-rw-r--r--	softmax_op_cudnn.cc	3791	log plain
-rw-r--r--	softmax_ops.cu	21595	log plain
-rw-r--r--	softmax_shared.cc	1392	log plain
-rw-r--r--	softmax_shared.h	462	log plain
-rw-r--r--	softmax_with_loss_op.cc	9068	log plain
-rw-r--r--	softmax_with_loss_op.h	2487	log plain
-rw-r--r--	softplus_op.cc	2067	log plain
-rw-r--r--	softplus_op.cu	1490	log plain
-rw-r--r--	softplus_op.h	781	log plain
-rw-r--r--	softsign_op.cc	2525	log plain
-rw-r--r--	softsign_op.cu	1436	log plain
-rw-r--r--	space_batch_op.cc	1780	log plain
-rw-r--r--	space_batch_op.h	6866	log plain
-rw-r--r--	space_batch_op_gpu.cu	4885	log plain
-rw-r--r--	sparse_normalize_op.cc	2217	log plain
-rw-r--r--	sparse_normalize_op.h	1034	log plain
-rw-r--r--	sparse_to_dense_mask_op.cc	4149	log plain
-rw-r--r--	sparse_to_dense_mask_op.h	9599	log plain
-rw-r--r--	sparse_to_dense_op.cc	1913	log plain
-rw-r--r--	sparse_to_dense_op.cu	2360	log plain
-rw-r--r--	sparse_to_dense_op.h	3816	log plain
-rw-r--r--	spatial_batch_norm_gradient_op.cc	5723	log plain
-rw-r--r--	spatial_batch_norm_op.cc	10714	log plain
-rw-r--r--	spatial_batch_norm_op.h	2606	log plain
-rw-r--r--	spatial_batch_norm_op_cudnn.cc	12373	log plain
-rw-r--r--	spatial_softmax_with_loss_op.cc	8134	log plain
-rw-r--r--	spatial_softmax_with_loss_op.h	2167	log plain
-rw-r--r--	sqrt_op.cc	1604	log plain
-rw-r--r--	sqrt_op.cu	726	log plain
-rw-r--r--	square_root_divide_op.cc	1083	log plain
-rw-r--r--	square_root_divide_op.h	1768	log plain
-rw-r--r--	stats_ops.cc	9378	log plain
-rw-r--r--	stop_gradient.cc	551	log plain
-rw-r--r--	stop_gradient.h	543	log plain
-rw-r--r--	stop_gradient_gpu.cc	191	log plain
-rw-r--r--	string_ops.cc	6138	log plain
-rw-r--r--	string_ops.h	2030	log plain
-rw-r--r--	string_ops_test.cc	4072	log plain
-rw-r--r--	stump_func_op.cc	2697	log plain
-rw-r--r--	stump_func_op.cu	1568	log plain
-rw-r--r--	stump_func_op.h	2078	log plain
-rw-r--r--	stylizer_ops.cc	18993	log plain
-rw-r--r--	summarize_op.cc	2289	log plain
-rw-r--r--	summarize_op.cu	3412	log plain
-rw-r--r--	summarize_op.h	1865	log plain
-rw-r--r--	swish_op.cc	2223	log plain
-rw-r--r--	swish_op.cu	1890	log plain
-rw-r--r--	swish_op.h	808	log plain
-rw-r--r--	tanh_op.cc	1989	log plain
-rw-r--r--	tanh_op.cu	1461	log plain
-rw-r--r--	tensor_protos_db_input.cc	1701	log plain
-rw-r--r--	tensor_protos_db_input.h	3715	log plain
-rw-r--r--	tensor_protos_db_input_gpu.cc	248	log plain
-rw-r--r--	text_file_reader.cc	6348	log plain
-rw-r--r--	text_file_reader_utils.cc	3093	log plain
-rw-r--r--	text_file_reader_utils.h	2830	log plain
-rw-r--r--	text_file_reader_utils_test.cc	4243	log plain
-rw-r--r--	thresholded_relu_op.cc	2800	log plain
-rw-r--r--	thresholded_relu_op.cu	1575	log plain
-rw-r--r--	thresholded_relu_op.h	1105	log plain
-rw-r--r--	tile_op.cc	2577	log plain
-rw-r--r--	tile_op.cu	2507	log plain
-rw-r--r--	tile_op.h	8273	log plain
-rw-r--r--	top_k.cc	8312	log plain
-rw-r--r--	top_k.cu	10295	log plain
-rw-r--r--	top_k.h	1086	log plain
-rw-r--r--	top_k_heap_selection.cuh	8535	log plain
-rw-r--r--	top_k_radix_selection.cuh	15280	log plain
-rw-r--r--	transpose_op.cc	3040	log plain
-rw-r--r--	transpose_op.cu	185	log plain
-rw-r--r--	transpose_op.h	2031	log plain
-rw-r--r--	transpose_op_cudnn.cc	4260	log plain
-rw-r--r--	tt_linear_op.cc	2569	log plain
-rw-r--r--	tt_linear_op.h	6205	log plain
-rw-r--r--	unique_ops.cc	3540	log plain
-rw-r--r--	unique_ops.cu	4296	log plain
-rw-r--r--	unique_ops.h	1641	log plain
-rw-r--r--	utility_ops.cc	32703	log plain
-rw-r--r--	utility_ops.cu	12475	log plain
-rw-r--r--	utility_ops.h	44230	log plain
-rw-r--r--	utility_ops_gpu.cc	2863	log plain
-rw-r--r--	utility_ops_gpu_test.cc	2458	log plain
-rw-r--r--	utility_ops_test.cc	2145	log plain
-rw-r--r--	variable_length_sequence_padding.cc	713	log plain
-rw-r--r--	variable_length_sequence_padding.h	1335	log plain
-rw-r--r--	weighted_multi_sampling_op.cc	3963	log plain
-rw-r--r--	weighted_multi_sampling_op.h	586	log plain
-rw-r--r--	weighted_sample_op.cc	4160	log plain
-rw-r--r--	weighted_sample_op.cu	2815	log plain
-rw-r--r--	weighted_sample_op.h	733	log plain
-rw-r--r--	while_op.cc	841	log plain
-rw-r--r--	while_op.h	1945	log plain
-rw-r--r--	while_op_gpu.cc	174	log plain
-rw-r--r--	workspace_ops.cc	1207	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