File List

Here is a list of all documented files with brief descriptions:

[detail level 1234567]

▼ binaries
caffe2_benchmark.cc
convert_caffe_image_db.cc
convert_db.cc
convert_encoded_to_raw_leveldb.cc
core_overhead_benchmark.cc
db_throughput.cc
inspect_gpus.cc
make_cifar_db.cc
make_image_db.cc
make_mnist_db.cc
predictor_verifier.cc
print_core_object_sizes.cc
print_registered_core_operators.cc
run_plan.cc
run_plan_mpi.cc
speed_benchmark.cc
split_db.cc
tutorial_blob.cc
zmq_feeder.cc
▼ caffe2
► core
► nomnigraph
► Converters
Caffe2.cc
► include
► nomnigraph
► Converters
Caffe2.h
Dot.h
► Graph
Algorithms.h
Graph.h
TarjansImpl.h
► Representations
Compiler.h
ControlFlow.h
NeuralNet.h
► Support
Casting.h
Common.h
Pointer.h
► Transformations
ConnectNet.h
Match.h
OperatorFusion.h
► Representations
NeuralNet.cc
► Transformations
ConnectNet.cc
OperatorFusion.cc
test.cc
allocator.cc
allocator.h
asan.h
blob.h
blob_serialization.cc
blob_serialization.h
blob_serialization_gpu.cc
blob_serializer_base.h
blob_stats.cc
blob_stats.h
common.cc
common.h
common_cudnn.cc
common_cudnn.h
common_gpu.cc
common_gpu.h
common_omp.h
context.cc
context.h
context_gpu.h
cudnn_wrappers.h
db.cc
db.h
event.cc
event.h
event_cpu.h
event_gpu.cc
flags.cc
flags.h	Commandline flags support for Caffe2
graph.cc
graph.h
init.cc
init.h
init_intrinsics_check.cc
init_omp.cc
logging.cc
logging.h
logging_is_google_glog.h
logging_is_not_google_glog.h
macros.h
memonger.cc
memonger.h
module.cc
module.h
net.cc
net.h
net_async_base.cc
net_async_base.h
net_async_dag_gpu.cc
net_async_dag_gpu.h
net_async_gpu_thread_pool.h
net_async_gpu_thread_pool_gpu.cc
net_async_polling.cc
net_async_polling.h
net_async_scheduling.cc
net_async_scheduling.h
net_dag.cc
net_dag.h
net_dag_utils.cc
net_dag_utils.h
net_simple.cc
net_simple.h
net_simple_async.cc
net_simple_async.h
net_singlethread_async_gpu.cc
numa.cc
numa.h
observer.h
operator.cc
operator.h
operator_gradient.h
operator_schema.cc
operator_schema.h
plan_executor.cc
plan_executor.h
predictor.cc
predictor.h
qtensor.cc
qtensor.h
qtensor_serialization.cc
qtensor_serialization.h
registry.h
scope_guard.h
static_tracepoint.h
static_tracepoint_elfx86.h
stats.cc
stats.h
tensor.cc
tensor.h
timer.h
transform.cc
transform.h
typeid.cc
typeid.h
types.cc
types.h
workspace.cc
workspace.h
► cuda_rtc
common_rtc.h
elemenntwise_rtc_gpu.cc
pool_op_rtc_gpu.cc
► db
create_db_op.cc
create_db_op.h
create_db_op_gpu.cc
leveldb.cc
lmdb.cc
protodb.cc
zmqdb.cc
► distributed
file_store_handler.cc
file_store_handler.h
file_store_handler_op.cc
file_store_handler_op.h
file_store_handler_op_gpu.cc
py_export.cc
redis_store_handler.cc
redis_store_handler.h
redis_store_handler_op.cc
redis_store_handler_op.h
redis_store_handler_op_gpu.cc
store_handler.cc
store_handler.h
store_ops.cc
store_ops.h
► experiments
► operators
fully_connected_op_decomposition.cc
fully_connected_op_decomposition.h
fully_connected_op_decomposition_gpu.cc
fully_connected_op_prune.cc
fully_connected_op_prune.h
fully_connected_op_sparse.cc
fully_connected_op_sparse.h
funhash_op.cc
funhash_op.h
sparse_funhash_op.cc
sparse_funhash_op.h
sparse_matrix_reshape_op.cc
sparse_matrix_reshape_op.h
tt_contraction_op.cc
tt_contraction_op.h
tt_contraction_op_gpu.cc
tt_pad_op.cc
tt_pad_op.h
► image
image_input_op.cc
image_input_op.h
image_input_op_gpu.cc
transform_gpu.h
► mkl
► utils
mkl_context.cc
mkl_context.h
mkl_dnn_cppwrapper.h
mkl_memory.cc
mkl_memory.h
mkl_operator.h
mkl_version_check.h
sgemm_pack.h
mkl_utils.h
mklmemory_serialization.cc
► mobile
► contrib
► arm-compute
► core
context.cc
context.h
net_gl.cc
net_gl.h
operator.cc
operator.h
rewrite_net.cc
rewrite_net.h
► operators
activation_ops.cc
activation_ops.h
concat_op.cc
conv_op.cc
copy_op.cc
elementwise_sum_op.cc
fully_connected_op.cc
norm_planar_yuv_op.cc
pool_op.cc
reshape_op.cc
resize_op.cc
softmax_op.cc
spatial_batch_norm_op.cc
► test
gl_model_test.h
gl_operator_test.h
► ios
► mpscnn
mpscnn.h
mpscnn_context.h
mpscnn_graph_mask.h
mpscnn_kernels.h
mpscnn_test.h
ios_caffe.cc
ios_caffe.h
ios_caffe_defines.h
ios_caffe_predictor.cc
ios_caffe_predictor.h
► libopencl-stub
► include
► CL
cl.h
cl_ext.h
cl_gl.h
cl_gl_ext.h
cl_platform.h
opencl.h
libopencl.h
► src
libopencl.c
► libvulkan-stub
► include
► vulkan
vk_platform.h
vulkan.h
libvulkan-stub.h
► src
libvulkan-stub.c
► nnapi
dlnnapi.c
dlnnapi.h
NeuralNetworks.h
nnapi.cc
nnapi.h
nnapi_benchmark.cc
► opengl
► android
AndroidGLContext.cc
AndroidGLContext.h
arm_neon_support.h
gl3stub.c
gl3stub.h
GLContext.cc
GLImageAllocator.cc
► core
arm_neon_support.h
DataTransfer.cc
DataTransfer.h
GL.h
GLContext.cc
GLContext.h
GLFilter.cc
GLFilter.h
GLImage.cc
GLImage.h
GLImageAllocator.cc
GLImageAllocator.h
GLLogging.h
GLPBO.cc
GLPBO.h
GLPlainTexture.cc
GLPlainTexture.h
GLPredictor.cc
GLPredictor.h
GLTexture.cc
GLTexture.h
ImageAllocator.h
rewrite_net.cc
rewrite_net.h
► ios
GLContext.cc
GLImageAllocator.cc
IOSGLContext.h
IOSGLImageAllocator.cc
IOSGLImageAllocator.h
IOSGLTexture.h
► operators
gl_tiling_utils.h
GLAdd.cc
GLConcat.cc
GLConvolution.cc
GLConvolution.h
GLCopyOps.cc
GLInstanceNorm.cc
GLMul.cc
GLNormPlanarYUV.cc
GLPadImage.cc
GLPool.cc
GLPRelu.cc
GLResize.cc
GLSigmoid.cc
GLSoftmax.cc
GLStylizer.cc
GLSub.cc
► test
opengl_test.h
TestGLConvolution.cc
TestGLConvolution.h
► snpe
snpe_ffi.cc
snpe_ffi.h
snpe_globals.cc
snpe_op.cc
snpe_op_benchmark.cc
► ulp2
ulp.cc
ulp.h
ulp_neon.cc
ulp_neon.h
► mpi
mpi_common.cc
mpi_common.h
► observers
operator_attaching_net_observer.h
profile_observer.h
profile_observer_gpu.cc
runcnt_observer.cc
runcnt_observer.h
time_observer.cc
time_observer.h
► onnx
backend.cc
backend.h
backend_rep.cc
backend_rep.h
device.cc
device.h
helper.cc
helper.h
onnx_exporter.cc
onnx_exporter.h
► operators
► rnn
recurrent_network_blob_fetcher_op.cc
recurrent_network_blob_fetcher_op.h
recurrent_network_blob_fetcher_op_gpu.cc
recurrent_network_executor.cc
recurrent_network_executor.h
recurrent_network_executor_gpu.cc
recurrent_network_executor_gpu.h
recurrent_network_executor_incl.h
recurrent_network_op.cc
recurrent_network_op.h
recurrent_op_cudnn.cc
recurrent_op_cudnn.h
rnn_capable_operator_observer.h
abs_op.cc
accumulate_op.cc
accumulate_op.h
accuracy_op.cc
accuracy_op.h
apmeter_op.cc
apmeter_op.h
arg_max_op.cc
arg_max_op.h
arg_ops.cc
arg_ops.h
arg_ops_eigen.h
assert_op.cc
assert_op.h
atomic_ops.cc
batch_box_cox_op.cc
batch_box_cox_op.h
batch_gather_ops.cc
batch_gather_ops.h
batch_matmul_op.cc
batch_matmul_op.h
batch_sparse_to_dense_op.cc
batch_sparse_to_dense_op.h
bbox_transform_op.cc
bbox_transform_op.h
boolean_mask_ops.cc
boolean_mask_ops.h
boolean_unmask_ops.cc
boolean_unmask_ops.h
box_with_nms_limit_op.cc
box_with_nms_limit_op.h
cast_op.cc
cast_op.h
ceil_op.cc
ceil_op.h
channel_backprop_stats_op.cc
channel_backprop_stats_op.h
channel_shuffle_op.cc
channel_shuffle_op.h
channel_stats_op.cc
channel_stats_op.h
clip_op.cc
clip_op.h
collect_and_distribute_fpn_rpn_proposals_op.cc
collect_and_distribute_fpn_rpn_proposals_op.h
communicator_op.cc
communicator_op_gpu.cc
concat_split_op.cc
concat_split_op.h
concat_split_op_gpu.cc
conditional_op.cc
conditional_op.h
conv_gradient_op.cc
conv_op.cc
conv_op.h
conv_op_cache_cudnn.cc
conv_op_cache_cudnn.h
conv_op_cudnn.cc
conv_op_eigen.cc
conv_op_gpu.cc
conv_op_impl.h
conv_op_shared.cc
conv_op_shared.h
conv_op_shared_gpu.cc
conv_pool_op_base.h
conv_transpose_gradient_op.cc
conv_transpose_op.cc
conv_transpose_op.h
conv_transpose_op_cudnn.cc
conv_transpose_op_gpu.cc
conv_transpose_op_impl.h
conv_transpose_op_mobile.cc
conv_transpose_op_mobile.h
conv_transpose_op_mobile_impl.h
conv_transpose_unpool_op_base.h
cos_op.cc
cosine_embedding_criterion_op.cc
cosine_embedding_criterion_op.h
counter_ops.cc
counter_ops.h
counter_ops_gpu.cc
create_scope_op.cc
create_scope_op.h
cross_entropy_op.cc
cross_entropy_op.h
dataset_ops.cc
dataset_ops.h
deform_conv_gradient_op.cc
deform_conv_op.cc
deform_conv_op.h
deform_conv_op_impl.h
distance_op.cc
distance_op.h
do_op.cc
do_op.h
do_op_gpu.cc
dropout_op.cc
dropout_op.h
dropout_op_cudnn.cc
elementwise_add_op.cc
elementwise_div_op.cc
elementwise_linear_op.cc
elementwise_linear_op.h
elementwise_logical_ops.cc
elementwise_logical_ops.h
elementwise_mul_op.cc
elementwise_op.cc
elementwise_op.h
elementwise_op_schema.cc
elementwise_op_test.h
elementwise_sub_op.cc
elementwise_sum_op.cc
elu_op.cc
elu_op.h
exp_op.cc
expand_squeeze_dims_op.cc
expand_squeeze_dims_op.h
expand_squeeze_dims_op_gpu.cc
extend_tensor_op.cc
fc_inference.cc
fc_inference.h
feed_blob_op.cc
feed_blob_op.h
filler_op.cc
filler_op.h
filler_op_gpu.cc
find_duplicate_elements_op.cc
find_duplicate_elements_op.h
find_op.cc
find_op.h
flatten_op.cc
flatten_op.h
flexible_top_k.cc
flexible_top_k.h
floor_op.cc
floor_op.h
free_op.cc
free_op.h
free_op_gpu.cc
fully_connected_op.cc
fully_connected_op.h
fully_connected_op_gpu.cc
fused_rowwise_8bit_conversion_ops.cc
fused_rowwise_8bit_conversion_ops.h
gather_fused_8bit_rowwise_op.cc
gather_fused_8bit_rowwise_op.h
gather_ranges_to_dense_op.cc
gather_ranges_to_dense_op.h
generate_proposals_op.cc
generate_proposals_op.h
generate_proposals_op_util_boxes.h
generate_proposals_op_util_nms.h
given_tensor_fill_op.cc
given_tensor_fill_op.h
glu_op.cc
glu_op.h
gru_unit_op.cc
gru_unit_op.h
h_softmax_op.cc
h_softmax_op.h
half_float_ops.cc
half_float_ops.h
if_op.cc
if_op.h
if_op_gpu.cc
im2col_op.cc
im2col_op.h
im2col_op_gpu.cc
index_hash_ops.cc
index_hash_ops.h
index_ops.cc
instance_norm_gradient_op.cc
instance_norm_op.cc
instance_norm_op.h
jsd_op.cc
jsd_op.h
key_split_ops.cc
key_split_ops.h
last_n_window_collector.cc
layer_norm_op.cc
layer_norm_op.h
leaky_relu_op.cc
leaky_relu_op.h
lengths_reducer_fused_8bit_rowwise_ops.cc
lengths_reducer_fused_8bit_rowwise_ops.h
lengths_reducer_ops.cc
lengths_reducer_ops.h
lengths_reducer_rowwise_8bit_ops.cc
lengths_reducer_rowwise_8bit_ops.h
lengths_tile_op.cc
lengths_tile_op.h
lengths_tile_op_gpu.cc
lengths_top_k_op.cc
lengths_top_k_op.h
listwise_l2r_op.cc
listwise_l2r_op.h
load_save_op.cc
load_save_op.h
load_save_op_gpu.cc
local_response_normalization_op.cc
local_response_normalization_op.h
local_response_normalization_op_cudnn.cc
locally_connected_op.cc
locally_connected_op.h
locally_connected_op_gpu.cc
locally_connected_op_impl.h
locally_connected_op_util.cc
locally_connected_op_util.h
log_op.cc
logit_op.cc
logit_op.h
loss_op.cc
loss_op.h
lp_pool_op.cc
lpnorm_op.cc
lpnorm_op.h
lstm_unit_op.cc
lstm_unit_op.h
map_ops.cc
map_ops.h
margin_ranking_criterion_op.cc
margin_ranking_criterion_op.h
math_ops.cc
math_ops.h
matmul_op.cc
matmul_op.h
matmul_op_gpu.cc
max_pool_with_index.h
mean_op.cc
mean_op.h
merge_id_lists_op.cc
merge_id_lists_op.h
minmax_gradient_ops.cc
minmax_ops.cc
minmax_ops.h
mod_op.cc
mod_op.h
multi_class_accuracy_op.cc
multi_class_accuracy_op.h
negate_gradient_op.cc
negate_gradient_op.h
negate_gradient_op_gpu.cc
negative_op.cc
ngram_ops.cc
ngram_ops.h
no_default_engine_op.h
norm_planar_yuv_op.cc
normalize_l1_op.cc
normalize_l1_op.h
normalize_op.cc
normalize_op.h
one_hot_ops.cc
one_hot_ops.h
onnx_while_op.cc
onnx_while_op.h
op_utils_cudnn.h
operator_fallback_gpu.h
order_switch_ops.cc
order_switch_ops.h
pack_rnn_sequence_op.cc
pack_rnn_sequence_op.h
pack_segments.cc
pack_segments.h
pad_op.cc
pad_op.h
partition_ops.cc
partition_ops.h
percentile_op.cc
percentile_op.h
perplexity_op.cc
perplexity_op.h
piecewise_linear_transform_op.cc
piecewise_linear_transform_op.h
pool_gradient_op.cc
pool_op.cc
pool_op.h
pow_op.cc
pow_op.h
prefetch_op.h
prelu_op.cc
prelu_op.h
prepend_dim_op.cc
prepend_dim_op.h
prepend_dim_op_gpu.cc
quant_decode_op.cc
quant_decode_op.h
rank_loss_op.cc
rank_loss_op.h
reduce_ops.cc
reduce_ops.h
reducer_functors.h
reduction_front_back_ops.cc
reduction_front_back_ops.h
reduction_ops.cc
reduction_ops.h
relu_op.cc
relu_op.h
relu_op_cudnn.cc
remove_data_blocks_op.cc
remove_data_blocks_op.h
replace_nan_op.cc
replace_nan_op.h
reservoir_sampling.cc
reshape_op.cc
reshape_op.h
reshape_op_gpu.cc
resize_op.cc
resize_op.h
reverse_packed_segs_op.cc
reverse_packed_segs_op.h
rmac_regions_op.cc
rmac_regions_op.h
roi_align_gradient_op.cc
roi_align_gradient_op.h
roi_align_op.cc
roi_align_op.h
roi_pool_op.cc
roi_pool_op.h
rowmul_op.cc
rowmul_op.h
scale_op.cc
scale_op.h
scale_op_gpu.cc
segment_reduction_op.cc
segment_reduction_op.h
selu_op.cc
selu_op.h
sequence_ops.cc
sequence_ops.h
shape_op.cc
shape_op.h
shape_op_gpu.cc
sigmoid_op.cc
sin_op.cc
sinusoid_position_encoding_op.cc
sinusoid_position_encoding_op.h
slice_op.cc
slice_op.h
softmax_op.cc
softmax_op.h
softmax_op_cudnn.cc
softmax_shared.cc
softmax_shared.h
softmax_with_loss_op.cc
softmax_with_loss_op.h
softplus_op.cc
softplus_op.h
softsign_op.cc
space_batch_op.cc
space_batch_op.h
sparse_normalize_op.cc
sparse_normalize_op.h
sparse_to_dense_mask_op.cc
sparse_to_dense_mask_op.h
sparse_to_dense_op.cc
sparse_to_dense_op.h
spatial_batch_norm_gradient_op.cc
spatial_batch_norm_op.cc
spatial_batch_norm_op.h
spatial_batch_norm_op_cudnn.cc
spatial_softmax_with_loss_op.cc
spatial_softmax_with_loss_op.h
sqrt_op.cc
square_root_divide_op.cc
square_root_divide_op.h
stats_ops.cc
stop_gradient.cc
stop_gradient.h
stop_gradient_gpu.cc
string_ops.cc
string_ops.h
stump_func_op.cc
stump_func_op.h
stylizer_ops.cc
summarize_op.cc
summarize_op.h
swish_op.cc
swish_op.h
tanh_op.cc
tensor_protos_db_input.cc
tensor_protos_db_input.h
tensor_protos_db_input_gpu.cc
text_file_reader.cc
text_file_reader_utils.cc
text_file_reader_utils.h
thresholded_relu_op.cc
thresholded_relu_op.h
tile_op.cc
tile_op.h
top_k.cc
top_k.h
transpose_op.cc
transpose_op.h
transpose_op_cudnn.cc
tt_linear_op.cc
tt_linear_op.h
utility_ops.cc
utility_ops.h
utility_ops_gpu.cc
variable_length_sequence_padding.cc
variable_length_sequence_padding.h
weighted_multi_sampling_op.cc
weighted_multi_sampling_op.h
weighted_sample_op.cc
weighted_sample_op.h
while_op.cc
while_op.h
while_op_gpu.cc
workspace_ops.cc
zero_gradient_op.cc
zero_gradient_op.h
zero_gradient_op_gpu.cc
► perfkernels
common.h
common_avx.cc
common_avx2.cc
cvtsh_ss_bugfix.h
embedding_lookup.cc
embedding_lookup.h
embedding_lookup_avx2.cc
embedding_lookup_fused_8bit_rowwise_avx2.cc
fused_8bit_rowwise_embedding_lookup.cc
fused_8bit_rowwise_embedding_lookup.h
typed_axpy.cc
typed_axpy.h
typed_axpy_avx.cc
typed_axpy_avx2.cc
► queue
blobs_queue.cc
blobs_queue.h
blobs_queue_db.cc
blobs_queue_db.h
queue_ops.cc
queue_ops.h
queue_ops_gpu.cc
rebatching_queue.cc
rebatching_queue.h
rebatching_queue_ops.cc
rebatching_queue_ops.h
► sgd
adagrad_op.cc
adagrad_op.h
adam_op.cc
adam_op.h
clip_tensor_op.cc
clip_tensor_op.h
fp16_momentum_sgd_op.h
fp32_momentum_sgd_op.h
ftrl_op.cc
ftrl_op.h
iter_op.cc
iter_op.h
iter_op_gpu.cc
lars_op.cc
lars_op.h
learning_rate_functors.h
learning_rate_op.cc
learning_rate_op.h
learning_rate_op_gpu.cc
momentum_sgd_op.cc
momentum_sgd_op.h
rmsprop_op.cc
rmsprop_op.h
yellowfin_op.cc
yellowfin_op.h
► share
► contrib
► nnpack
conv_op.cc
► zstd
quant_decomp_zstd_op.cc
quant_decomp_zstd_op.h
► transforms
common_subexpression_elimination.cc
common_subexpression_elimination.h
conv_to_nnpack_transform.cc
conv_to_nnpack_transform.h
pattern_net_transform.cc
pattern_net_transform.h
single_op_transform.cc
single_op_transform.h
► utils
► threadpool
pthreadpool.cc
pthreadpool.h
pthreadpool_impl.cc
pthreadpool_impl.h
ThreadPool.cc
ThreadPool.h
ThreadPoolCommon.h
WorkersPool.h
cast.h
cblas.h
conversions.h
cpu_neon.h
cpuid.cc
cpuid.h
eigen_utils.h
fixed_divisor.h
map_utils.h
math-detail.h
math.h
math_cpu.cc
mixed_utils.h
murmur_hash3.cc
murmur_hash3.h
proto_utils.cc
proto_utils.h
signal_handler.cc
signal_handler.h
simple_queue.h
smart_tensor_printer.cc
smart_tensor_printer.h
string_utils.cc
string_utils.h
thread_pool.h
zmq_helper.h
► video
optical_flow.cc
optical_flow.h
video_decoder.cc
video_decoder.h
video_input_op.cc
video_input_op.h
video_input_op_gpu.cc
video_io.cc
video_io.h
▼ modules
► detectron
affine_channel_op.cc
affine_channel_op.h
batch_permutation_op.cc
batch_permutation_op.h
group_spatial_softmax_op.cc
group_spatial_softmax_op.h
ps_roi_pool_op.cc
ps_roi_pool_op.h
roi_pool_f_op.cc
roi_pool_f_op.h
sample_as_op.cc
sample_as_op.h
select_smooth_l1_loss_op.cc
select_smooth_l1_loss_op.h
sigmoid_cross_entropy_loss_op.cc
sigmoid_cross_entropy_loss_op.h
sigmoid_focal_loss_op.cc
sigmoid_focal_loss_op.h
smooth_l1_loss_op.cc
smooth_l1_loss_op.h
softmax_focal_loss_op.cc
softmax_focal_loss_op.h
spatial_narrow_as_op.cc
spatial_narrow_as_op.h
upsample_nearest_op.cc
upsample_nearest_op.h
► module_test
module_test_dynamic.cc
► observers
net_observer_reporter.h
net_observer_reporter_print.cc
net_observer_reporter_print.h
observer_config.cc
observer_config.h
perf_observer.cc
perf_observer.h
► rocksdb
rocksdb.cc