▼ 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 | |