File List

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

[detail level 12345]

► aten
► src
► ATen
code_template.py
common_with_cwrap.py
cwrap_parser.py
extract_cwrap.py
function_wrapper.py
gen.py
native_parse.py
nn_parse.py
preprocess_declarations.py
► binaries
► bench_gen
bench_gen.py
► caffe2
► distributed
__init__.py
store_ops_test_util.py
► experiments
► python
__init__.py
convnet_benchmarks.py
device_reduce_sum_bench.py
net_construct_bench.py
SparseTransformer.py
__init__.py
► python
► docs
__init__.py
formatter.py
generator.py
github.py
parser.py
► examples
__init__.py
char_rnn.py
lmdb_create_example.py
resnet50_trainer.py
► helpers
__init__.py
algebra.py
arg_scope.py
array_helpers.py
control_ops.py
conv.py
db_input.py
dropout.py
elementwise_linear.py
fc.py
nonlinearity.py
normalization.py
pooling.py
tools.py
train.py
► ideep
__init__.py
test_ideep_net.py
transform_ideep_net.py
► layers
__init__.py
adaptive_weight.py
add_bias.py
arc_cosine_feature_map.py
batch_distill_lr_loss.py
batch_lr_loss.py
batch_mse_loss.py
batch_normalization.py
batch_sigmoid_cross_entropy_loss.py
batch_softmax_loss.py
blob_weighted_sum.py
bucket_weighted.py
build_index.py
concat.py
constant_weight.py
conv.py
dropout.py
fc.py
fc_without_bias.py
feature_sparse_to_dense.py
functional.py
gather_record.py
homotopy_weight.py
label_smooth.py
last_n_window_collector.py
layer_normalization.py
layers.py
margin_rank_loss.py
merge_id_lists.py
pairwise_similarity.py
position_weighted.py
random_fourier_features.py
reservoir_sampling.py
sampling_train.py
sampling_trainable_mixin.py
select_record_by_context.py
semi_random_features.py
sparse_feature_hash.py
sparse_lookup.py
split.py
tags.py
uniform_sampling.py
► mint
__init__.py
app.py
► mkl
__init__.py
rewrite_graph.py
► modeling
__init__.py
compute_histogram_for_blobs.py
compute_norm_for_blobs.py
compute_statistics_for_blobs.py
get_entry_from_blobs.py
gradient_clipping.py
initializers.py
net_modifier.py
parameter_info.py
parameter_sharing.py
► models
► seq2seq
__init__.py
beam_search.py
seq2seq_model_helper.py
seq2seq_util.py
train.py
translate.py
__init__.py
__sym_init__.py
download.py
resnet.py
► onnx
► bin
__init__.py
conversion.py
► tests
__init__.py
test_utils.py
__init__.py
backend.py
backend_cpp_rep.py
backend_rep.py
error.py
frontend.py
helper.py
onnxifi.py
test_onnxifi.py
workspace.py
► operator_test
__init__.py
adagrad_test_helper.py
conftest.py
detectron_keypoints.py
sparse_lengths_sum_benchmark.py
► predictor
__init__.py
mobile_exporter.py
predictor_exporter.py
predictor_py_utils.py
serde.py
► rnn
__init__.py
lstm_comparison.py
rnn_cell_test_util.py
► serialized_test
__init__.py
coverage.py
serialized_test_util.py
► test
__init__.py
executor_test_util.py
► trt
__init__.py
test_trt.py
transform.py
__init__.py
_import_c_extension.py
attention.py
benchmark_generator.py
binarysize.py
brew.py
build.py
cached_reader.py
caffe_translator.py
checkpoint.py
cnn.py
compatibility.py
context.py
control.py
control_ops_grad.py
control_ops_util.py
convert.py
convnet_benchmarks.py
core.py
crf.py
crf_predict.py
data_parallel_model.py
data_workers.py
dataio.py
dataset.py
db_file_reader.py
device_checker.py
dyndep.py
embedding_generation_benchmark.py
experiment_util.py
extension_loader.py
functional.py
gradient_checker.py
gru_cell.py
hip_test_util.py
hsm_util.py
hypothesis_test_util.py
ideep_test_util.py
layer_model_helper.py
layer_model_instantiator.py
layer_test_util.py
lstm_benchmark.py
memonger.py
mkl_test_util.py
model_helper.py
modifier_context.py
muji.py
net_builder.py
net_drawer.py
net_printer.py
nomnigraph.py
nomnigraph_transformations.py
normalizer.py
normalizer_context.py
numa_benchmark.py
optimizer.py
optimizer_context.py
optimizer_test_util.py
parallel_workers.py
pipeline.py
predictor_constants.py
queue_util.py
record_queue.py
recurrent.py
regularizer.py
regularizer_context.py
rnn_cell.py
schema.py
scope.py
session.py
task.py
test_util.py
text_file_reader.py
timeout_guard.py
transformations.py
tt_core.py
utils.py
visualize.py
workspace.py
► quantization
► server
__init__.py
dnnlowp_test_utils.py
utils.py
__init__.py
__init__.py
► scripts
► model_zoo
update-caffe2-models.py
update-models-from-caffe2.py
diagnose_protobuf.py
get_python_cmake_flags.py
► test
► bottleneck
test.py
test_args.py
test_cuda.py
► cpp
► api
init_baseline.py
optim_baseline.py
► jit
__init__.py
tests_setup.py
__init__.py
► cpp_extensions
► no_python_abi_suffix_test
setup.py
► torch_test_cpp_extension
__init__.py
setup.py
► custom_operator
model.py
test_custom_ops.py
► data
network1.py
network2.py
► error_messages
storage.py
► onnx
► model_defs
__init__.py
dcgan.py
lstm_flattening_result.py
mnist.py
rnn_model_with_packed_sequence.py
squeezenet.py
srresnet.py
super_resolution.py
word_language_model.py
debug_embed_params.py
export_onnx_tests_filter.py
export_onnx_tests_generator.py
pytorch_helper.py
test_caffe2_common.py
test_models.py
test_onnx_common.py
test_operators.py
test_pytorch_common.py
test_pytorch_helper.py
test_pytorch_onnx_caffe2.py
test_verify.py
verify.py
► optim
test.py
► test_module
__init__.py
future_div.py
no_future_div.py
common_cuda.py
common_methods_invocations.py
common_nn.py
common_utils.py
expecttest.py
test_autograd.py
test_c10d.py
test_cpp_extensions.py
test_cuda.py
test_cuda_primary_ctx.py
test_dataloader.py
test_distributed.py
test_distributions.py
test_docs_coverage.py
test_expecttest.py
test_indexing.py
test_indexing_cuda.py
test_jit.py
test_multiprocessing.py
test_multiprocessing_spawn.py
test_namedtuple_return_api.py
test_nccl.py
test_nn.py
test_numba_integration.py
test_optim.py
test_sparse.py
test_thd_distributed.py
test_torch.py
test_type_hints.py
test_type_info.py
test_utils.py
► tools
► amd_build
► pyHIPIFY
__init__.py
constants.py
cuda_to_hip_mappings.py
hipify_python.py
build_amd.py
► autograd
__init__.py
gen_autograd.py
gen_autograd_functions.py
gen_python_functions.py
gen_variable_factories.py
gen_variable_type.py
load_derivatives.py
nested_dict.py
utils.py
► cwrap
► plugins
__init__.py
ArgcountChecker.py
ArgumentReferences.py
AutoGPU.py
BeforeAfterCall.py
ConstantArguments.py
CuDNNPlugin.py
GILRelease.py
NNExtension.py
NullableArguments.py
OptionalArguments.py
ReturnArguments.py
WrapDim.py
__init__.py
cwrap.py
► jit
__init__.py
gen_jit_dispatch.py
► nnwrap
__init__.py
generate_wrappers.py
► pyi
__init__.py
gen_pyi.py
► setup_helpers
__init__.py
cuda.py
cudnn.py
dist_check.py
env.py
generate_code.py
miopen.py
nccl.py
nnpack.py
nvtoolext.py
qnnpack.py
rocm.py
split_types.py
► shared
__init__.py
module_loader.py
__init__.py
build_libtorch.py
build_pytorch_libs.py
build_variables.py
clang_format.py
clang_tidy.py
download_mnist.py
flake8_hook.py
► torch
► _thnn
__init__.py
utils.py
► autograd
► _functions
__init__.py
tensor.py
utils.py
__init__.py
anomaly_mode.py
function.py
grad_mode.py
gradcheck.py
profiler.py
variable.py
► backends
► cuda
__init__.py
► cudnn
__init__.py
rnn.py
► mkl
__init__.py
► openmp
__init__.py
__init__.py
► contrib
__init__.py
_graph_vis.py
_tensorboard_vis.py
► cuda
__init__.py
_utils.py
comm.py
error.py
nccl.py
nvtx.py
profiler.py
random.py
sparse.py
streams.py
► distributed
► deprecated
__init__.py
remote_types.py
__init__.py
distributed_c10d.py
launch.py
rendezvous.py
► distributions
__init__.py
bernoulli.py
beta.py
binomial.py
categorical.py
cauchy.py
chi2.py
constraint_registry.py
constraints.py
dirichlet.py
distribution.py
exp_family.py
exponential.py
fishersnedecor.py
gamma.py
geometric.py
gumbel.py
half_cauchy.py
half_normal.py
independent.py
kl.py
laplace.py
log_normal.py
logistic_normal.py
lowrank_multivariate_normal.py
multinomial.py
multivariate_normal.py
negative_binomial.py
normal.py
one_hot_categorical.py
pareto.py
poisson.py
relaxed_bernoulli.py
relaxed_categorical.py
studentT.py
transformed_distribution.py
transforms.py
uniform.py
utils.py
weibull.py
► for_onnx
__init__.py
► jit
__init__.py
annotations.py
batchop.py
frontend.py
quantized.py
supported_ops.py
► lib
► THD
► benchmark
benchmark.py
► multiprocessing
__init__.py
pool.py
queue.py
reductions.py
spawn.py
► nn
► _functions
► thnn
__init__.py
auto.py
auto_double_backwards.py
auto_symbolic.py
normalization.py
sparse.py
__init__.py
vision.py
► backends
__init__.py
backend.py
thnn.py
► modules
__init__.py
_functions.py
activation.py
adaptive.py
batchnorm.py
container.py
conv.py
distance.py
dropout.py
fold.py
instancenorm.py
linear.py
loss.py
module.py
normalization.py
padding.py
pixelshuffle.py
pooling.py
rnn.py
sparse.py
upsampling.py
utils.py
► parallel
► deprecated
__init__.py
distributed.py
distributed_cpu.py
__init__.py
_functions.py
data_parallel.py
distributed.py
distributed_cpu.py
parallel_apply.py
replicate.py
scatter_gather.py
► utils
__init__.py
clip_grad.py
convert_parameters.py
rnn.py
spectral_norm.py
sync_batch_norm.py
weight_norm.py
__init__.py
_reduction.py
_VF.py
cpp.py
functional.py
grad.py
init.py
parameter.py
► onnx
__init__.py
operators.py
symbolic.py
utils.py
► optim
__init__.py
adadelta.py
adagrad.py
adam.py
adamax.py
asgd.py
lbfgs.py
lr_scheduler.py
optimizer.py
rmsprop.py
rprop.py
sgd.py
sparse_adam.py
► sparse
__init__.py
► testing
__init__.py
► utils
► backcompat
__init__.py
► bottleneck
__init__.py
__main__.py
► data
► _utils
__init__.py
collate.py
pin_memory.py
signal_handling.py
worker.py
__init__.py
dataloader.py
dataset.py
distributed.py
sampler.py
► ffi
__init__.py
__init__.py
_cpp_extension_versioner.py
checkpoint.py
collect_env.py
cpp_extension.py
dlpack.py
file_baton.py
hooks.py
model_zoo.py
__init__.py
_jit_internal.py
_ops.py
_six.py
_storage_docs.py
_tensor_docs.py
_tensor_str.py
_torch_docs.py
_utils.py
_utils_internal.py
functional.py
hub.py
random.py
serialization.py
storage.py
tensor.py
setup.py