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

Mode	Name	Size
-rw-r--r--	BREW	1029	log plain
-rw-r--r--	CMakeLists.txt	1349	log plain
-rw-r--r--	accumulate_op.cc	1211	log plain
-rw-r--r--	accumulate_op.cu	222	log plain
-rw-r--r--	accumulate_op.h	1200	log plain
-rw-r--r--	accuracy_op.cc	1900	log plain
-rw-r--r--	accuracy_op.cu	1634	log plain
-rw-r--r--	accuracy_op.h	483	log plain
-rw-r--r--	atomic_ops.cc	4451	log plain
-rw-r--r--	batch_matmul_op.cc	3687	log plain
-rw-r--r--	batch_matmul_op.h	2340	log plain
-rw-r--r--	batch_matmul_op_gpu.cc	193	log plain
-rw-r--r--	boolean_mask_ops.cc	4480	log plain
-rw-r--r--	cast_op.cc	1864	log plain
-rw-r--r--	cast_op.cu	898	log plain
-rw-r--r--	cast_op.h	3620	log plain
-rw-r--r--	clip_op.cc	2216	log plain
-rw-r--r--	clip_op.cu	2146	log plain
-rw-r--r--	clip_op.h	1599	log plain
-rw-r--r--	communicator_op.cc	5126	log plain
-rw-r--r--	communicator_op_gpu.cc	700	log plain
-rw-r--r--	concat_split_op.cc	2231	log plain
-rw-r--r--	concat_split_op.h	6888	log plain
-rw-r--r--	concat_split_op_gpu.cc	417	log plain
-rw-r--r--	conv_op.cc	2648	log plain
-rw-r--r--	conv_op.h	2030	log plain
-rw-r--r--	conv_op_cache_cudnn.cc	1177	log plain
-rw-r--r--	conv_op_cache_cudnn.h	494	log plain
-rw-r--r--	conv_op_cache_cudnn_test.cc	1304	log plain
-rw-r--r--	conv_op_cudnn.cc	23892	log plain
-rw-r--r--	conv_op_eigen.cc	7774	log plain
-rw-r--r--	conv_op_gpu.cc	321	log plain
-rw-r--r--	conv_op_impl.h	18379	log plain
-rw-r--r--	conv_op_shared.cc	564	log plain
-rw-r--r--	conv_op_shared.h	402	log plain
-rw-r--r--	conv_op_shared_gpu.cc	396	log plain
-rw-r--r--	conv_pool_op_base.h	13646	log plain
-rw-r--r--	conv_transpose_op.cc	2712	log plain
-rw-r--r--	conv_transpose_op.h	1491	log plain
-rw-r--r--	conv_transpose_op_cudnn.cc	19835	log plain
-rw-r--r--	conv_transpose_op_gpu.cc	384	log plain
-rw-r--r--	conv_transpose_op_impl.h	14701	log plain
-rw-r--r--	conv_transpose_unpool_op_base.h	7244	log plain
-rw-r--r--	cosine_embedding_criterion_op.cc	2527	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	4830	log plain
-rw-r--r--	counter_ops.h	4415	log plain
-rw-r--r--	counter_ops_gpu.cc	592	log plain
-rw-r--r--	cross_entropy_op.cc	13526	log plain
-rw-r--r--	cross_entropy_op.cu	4513	log plain
-rw-r--r--	cross_entropy_op.h	2935	log plain
-rw-r--r--	dataset_ops.cc	36762	log plain
-rw-r--r--	distance_op.cc	5751	log plain
-rw-r--r--	distance_op.cu	1432	log plain
-rw-r--r--	distance_op.h	6529	log plain
-rw-r--r--	dropout_op.cc	3220	log plain
-rw-r--r--	dropout_op.cu	2658	log plain
-rw-r--r--	dropout_op.h	1406	log plain
-rw-r--r--	elementwise_op.cc	10232	log plain
-rw-r--r--	elementwise_op.cu	3985	log plain
-rw-r--r--	elementwise_op.h	8022	log plain
-rw-r--r--	elementwise_op_gpu_test.cc	960	log plain
-rw-r--r--	elementwise_op_schema.cc	9920	log plain
-rw-r--r--	elementwise_op_test.cc	767	log plain
-rw-r--r--	elementwise_op_test.h	8541	log plain
-rw-r--r--	elu_op.cc	2320	log plain
-rw-r--r--	elu_op.h	895	log plain
-rw-r--r--	exp_op.cc	1193	log plain
-rw-r--r--	exp_op.cu	782	log plain
-rw-r--r--	extend_tensor_op.cc	2455	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	4782	log plain
-rw-r--r--	filler_op.cu	1143	log plain
-rw-r--r--	filler_op.h	9687	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--	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	1622	log plain
-rw-r--r--	fully_connected_op.h	5487	log plain
-rw-r--r--	fully_connected_op_gpu.cc	328	log plain
-rw-r--r--	fully_connected_op_gpu_test.cc	1627	log plain
-rw-r--r--	fully_connected_op_test.cc	1416	log plain
-rw-r--r--	h_softmax_op.cc	22362	log plain
-rw-r--r--	h_softmax_op.h	4951	log plain
-rw-r--r--	half_float_ops.cu	2525	log plain
-rw-r--r--	im2col_op.cc	594	log plain
-rw-r--r--	im2col_op.h	8796	log plain
-rw-r--r--	im2col_op_gpu.cc	270	log plain
-rw-r--r--	index_ops.cc	13486	log plain
-rw-r--r--	instance_norm_op.cc	4814	log plain
-rw-r--r--	instance_norm_op.h	1383	log plain
-rw-r--r--	last_n_window_collector.cc	3506	log plain
-rw-r--r--	load_save_op.cc	3386	log plain
-rw-r--r--	load_save_op.h	10995	log plain
-rw-r--r--	load_save_op_gpu.cc	601	log plain
-rw-r--r--	local_response_normalization_op.cc	8969	log plain
-rw-r--r--	local_response_normalization_op.cu	10438	log plain
-rw-r--r--	local_response_normalization_op.h	2586	log plain
-rw-r--r--	loss_op.cc	1101	log plain
-rw-r--r--	loss_op.cu	1229	log plain
-rw-r--r--	loss_op.h	1669	log plain
-rw-r--r--	lp_pool_op.cc	9314	log plain
-rw-r--r--	lp_pool_op.cu	10004	log plain
-rw-r--r--	lstm_unit_op.cc	1093	log plain
-rw-r--r--	lstm_unit_op.h	5531	log plain
-rw-r--r--	lstm_unit_op_gpu.cu	3929	log plain
-rw-r--r--	margin_ranking_criterion_op.cc	3412	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--	matmul_op.cc	3514	log plain
-rw-r--r--	matmul_op.h	2327	log plain
-rw-r--r--	matmul_op_gpu.cc	178	log plain
-rw-r--r--	metrics_ops.cc	1720	log plain
-rw-r--r--	metrics_ops.h	2572	log plain
-rw-r--r--	multi_class_accuracy_op.cc	2292	log plain
-rw-r--r--	multi_class_accuracy_op.cu	2167	log plain
-rw-r--r--	multi_class_accuracy_op.h	539	log plain
-rw-r--r--	negative_op.cc	1137	log plain
-rw-r--r--	negative_op.cu	764	log plain
-rw-r--r--	no_default_engine_op.h	1051	log plain
-rw-r--r--	normalize_op.cc	2166	log plain
-rw-r--r--	normalize_op.h	870	log plain
-rw-r--r--	one_hot_ops.cc	3519	log plain
-rw-r--r--	operator_fallback_gpu.h	4226	log plain
-rw-r--r--	operator_fallback_gpu_test.cc	2550	log plain
-rw-r--r--	order_switch_ops.cc	2795	log plain
-rw-r--r--	order_switch_ops.cu	1778	log plain
-rw-r--r--	order_switch_ops.h	851	log plain
-rw-r--r--	pack_segments.cc	6225	log plain
-rw-r--r--	pad_op.cc	14938	log plain
-rw-r--r--	pad_op.h	2710	log plain
-rw-r--r--	pad_op_gpu.cu	15690	log plain
-rw-r--r--	partition_ops.cc	3521	log plain
-rw-r--r--	partition_ops.h	6839	log plain
-rw-r--r--	perplexity_op.cc	1271	log plain
-rw-r--r--	perplexity_op.cu	1202	log plain
-rw-r--r--	perplexity_op.h	447	log plain
-rw-r--r--	pool_op.cc	22621	log plain
-rw-r--r--	pool_op.cu	17040	log plain
-rw-r--r--	pool_op.h	1557	log plain
-rw-r--r--	pool_op_cudnn.cc	7484	log plain
-rw-r--r--	prefetch_op.h	3973	log plain
-rw-r--r--	prelu_op.cc	8597	log plain
-rw-r--r--	prelu_op.h	1004	log plain
-rw-r--r--	rank_loss_op.cc	4378	log plain
-rw-r--r--	rank_loss_op.h	813	log plain
-rw-r--r--	recurrent_network_op.cc	2120	log plain
-rw-r--r--	recurrent_network_op.h	20465	log plain
-rw-r--r--	recurrent_op_cudnn.cc	15177	log plain
-rw-r--r--	recurrent_op_cudnn.h	3872	log plain
-rw-r--r--	reducer_functors.h	20324	log plain
-rw-r--r--	relu_op.cc	2388	log plain
-rw-r--r--	relu_op.cu	1448	log plain
-rw-r--r--	relu_op.h	755	log plain
-rw-r--r--	relu_op_cudnn.cc	5363	log plain
-rw-r--r--	relu_op_fp16.cu	2709	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--	reverse_packed_segs_op.cc	1146	log plain
-rw-r--r--	reverse_packed_segs_op.h	2602	log plain
-rw-r--r--	rowmul_op.cc	1550	log plain
-rw-r--r--	rowmul_op.h	1907	log plain
-rw-r--r--	scale_op.cc	790	log plain
-rw-r--r--	scale_op.h	865	log plain
-rw-r--r--	scale_op_gpu.cc	179	log plain
-rw-r--r--	segment_reduction_op.cc	55790	log plain
-rw-r--r--	sequence_ops.cc	17936	log plain
-rw-r--r--	sigmoid_op.cc	1907	log plain
-rw-r--r--	sigmoid_op.cu	1486	log plain
-rw-r--r--	softmax_op.cc	3441	log plain
-rw-r--r--	softmax_op.cu	3722	log plain
-rw-r--r--	softmax_op.h	1157	log plain
-rw-r--r--	softmax_op_cudnn.cc	3223	log plain
-rw-r--r--	softmax_shared.cc	1355	log plain
-rw-r--r--	softmax_shared.h	450	log plain
-rw-r--r--	softmax_with_loss_op.cc	9402	log plain
-rw-r--r--	softmax_with_loss_op.cu	15048	log plain
-rw-r--r--	softmax_with_loss_op.h	1939	log plain
-rw-r--r--	softsign_op.cc	2451	log plain
-rw-r--r--	softsign_op.cu	1463	log plain
-rw-r--r--	space_batch_op.cc	1795	log plain
-rw-r--r--	space_batch_op.h	6858	log plain
-rw-r--r--	space_batch_op_gpu.cu	4899	log plain
-rw-r--r--	sparse_to_dense_mask_op.cc	6214	log plain
-rw-r--r--	spatial_batch_norm_op.cc	12203	log plain
-rw-r--r--	spatial_batch_norm_op.h	2266	log plain
-rw-r--r--	spatial_batch_norm_op_cudnn.cc	9432	log plain
-rw-r--r--	square_root_divide_op.cc	1101	log plain
-rw-r--r--	square_root_divide_op.h	1608	log plain
-rw-r--r--	stop_gradient.cc	504	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	4072	log plain
-rw-r--r--	string_ops.h	1211	log plain
-rw-r--r--	summarize_op.cc	2198	log plain
-rw-r--r--	summarize_op.cu	3440	log plain
-rw-r--r--	summarize_op.h	1569	log plain
-rw-r--r--	tanh_op.cc	1947	log plain
-rw-r--r--	tanh_op.cu	1489	log plain
-rw-r--r--	tensor_protos_db_input.cc	1729	log plain
-rw-r--r--	tensor_protos_db_input.h	3745	log plain
-rw-r--r--	tensor_protos_db_input_gpu.cc	276	log plain
-rw-r--r--	text_file_reader.cc	6177	log plain
-rw-r--r--	text_file_reader_utils.cc	3069	log plain
-rw-r--r--	text_file_reader_utils.h	2802	log plain
-rw-r--r--	text_file_reader_utils_test.cc	4243	log plain
-rw-r--r--	transpose_op.cc	2554	log plain
-rw-r--r--	transpose_op.cu	2506	log plain
-rw-r--r--	transpose_op.h	1971	log plain
-rw-r--r--	tt_linear_op.cc	2533	log plain
-rw-r--r--	tt_linear_op.h	6205	log plain
-rw-r--r--	utility_ops.cc	26093	log plain
-rw-r--r--	utility_ops.h	45998	log plain
-rw-r--r--	utility_ops_gpu.cc	1555	log plain
-rw-r--r--	utility_ops_gpu_test.cc	1743	log plain
-rw-r--r--	utility_ops_test.cc	1531	log plain
-rw-r--r--	workspace_ops.cc	1207	log plain