File tree

• .circleci
  • README.md
  • cimodel/data
    • binary_build_definitions.py
  • config.yml
  • scripts
    • binary_populate_env.sh
  • verbatim-sources
    • workflows-binary-builds-smoke-subset.yml
• .github/workflows
  • lint.yml
• .jenkins/pytorch
  • build.sh
  • test.sh
  • win-test-helpers
    • test_python_all_except_nn.bat
• CMakeLists.txt
• android
  • build.gradle
  • libs
    • fbjni
  • pytorch_android
    • CMakeLists.txt
    • host
      • build.gradle
      • gradle.properties
    • src/main/cpp
      • pytorch_jni_common.cpp
• aten/src
  • ATen
    • CMakeLists.txt
    • Declarations.cwrap
    • ParallelNative.cpp
    • SparseTensorImpl.cpp
    • SparseTensorUtils.h
    • Utils.h
    • core
      • LegacyTypeDispatch.h
      • dispatch
        • DispatchKeyExtractor.h
        • DispatchTable.h
        • Dispatcher.cpp
        • Dispatcher.h
        • OperatorEntry.h
      • function_schema.h
    • function_wrapper.py
    • native
      • LinearAlgebra.cpp
      • Pow.cpp
      • QuantizedLinear.cpp
      • Resize.cpp
      • ResizeCommon.h
      • TensorConversions.cpp
      • TensorIterator.cpp
      • TensorIterator.h
      • cuda
        • AdaptiveAveragePooling.cu
        • Resize.cu
      • mkldnn
        • MKLDNNCommon.cpp
      • native_functions.yaml
      • quantized/cpu
        • tensor_operators.cpp
      • sparse
        • SparseTensor.cpp
    • templates
      • TensorBody.h
    • test
      • tensor_interop_test.cpp
  • THC/generic
    • THCTensorMathReduce.cu
  • TH
    • THMath.h
• azure-pipelines.yml
• benchmarks/operator_benchmark
  • benchmark_all_test.py
  • benchmark_core.py
  • benchmark_pytorch.py
  • benchmark_runner.py
  • pt
    • add_test.py
    • as_strided_test.py
    • cat_test.py
    • conv_test.py
    • embeddingbag_test.py
    • linear_test.py
    • matmul_test.py
• binaries
  • dump_operator_names.cc
• c10/core
  • DefaultTensorOptions.h
  • ScalarType.h
  • TensorImpl.h
  • TensorOptions.h
• caffe2
  • CMakeLists.txt
  • contrib/pytorch
    • script_module_op.cc
  • core
    • tensor.cc
    • tensor.h
  • operators
    • load_save_op.h
    • load_save_op_util.cc
    • load_save_op_util.h
  • perfkernels
    • fused_8bit_rowwise_embedding_lookup.h
    • fused_8bit_rowwise_embedding_lookup_idx.h
  • proto
    • caffe2.proto
  • python
    • pybind_state.cc
  • quantization/server
    • dynamic_histogram.cc
    • dynamic_histogram.h
  • utils/threadpool
    • ThreadPool.cc
• cmake
  • Summary.cmake
  • TorchConfig.cmake.in
  • public
    • utils.cmake
• docs
  • cpp/source
    • installing.rst
  • source
    • index.rst
    • notes
      • autograd.rst
      • distributed_autograd.rst
    • quantization.rst
    • rpc.rst
• scripts
  • build_android.sh
  • build_ios.sh
  • build_mobile.sh
• setup.py
• test
  • backward_compatibility
    • check_backward_compatibility.py
  • common_nn.py
  • cpp_api_parity
    • parity-tracker.md
  • cpp
    • api
      • functional.cpp
      • integration.cpp
      • modulelist.cpp
      • modules.cpp
      • sequential.cpp
      • tensor.cpp
    • dist_autograd
      • test_dist_autograd.cpp
    • jit
      • test_autodiff.cpp
      • test_class_import.cpp
      • test_interface.cpp
      • test_lite_interpreter.cpp
      • tests.h
  • dist_autograd_test.py
  • dist_utils.py
  • jit
    • test_module_interface.py
  • mobile/op_deps
    • CMakeLists.txt
    • build.sh
    • expected_deps.yaml
    • main.cc
    • simple_ops.cpp
    • simple_ops.h
    • utils.cpp
    • utils.h
  • onnx
    • test_pytorch_onnx_onnxruntime.py
  • rpc_test.py
  • test_autograd.py
  • test_cuda.py
  • test_jit.py
  • test_jit_py3.py
  • test_nn.py
  • test_quantized_nn_mods.py
  • test_torch.py
  • test_type_promotion.py
• third_party
  • fbgemm
• tools
  • autograd/templates
    • VariableType.h
    • python_torch_functions.cpp
  • build_variables.py
  • run-clang-tidy-in-ci.sh
  • setup_helpers
    • cmake.py
• torch
  • _torch_docs.py
  • csrc
    • Dtype.cpp
    • api
      • include/torch
        • nn
          • functional.h
          • functional
            • activation.h
            • dropout.h
            • loss.h
            • normalization.h
          • modules
            • activation.h
            • batchnorm.h
            • container
              • modulelist.h
              • sequential.h
            • conv.h
            • dropout.h
            • embedding.h
            • linear.h
            • loss.h
            • normalization.h
          • options
            • activation.h
            • batchnorm.h
            • dropout.h
            • loss.h
            • normalization.h
        • optim
          • serialize.h
        • serialize.h
        • serialize
          • output-archive.h
      • src
        • nn
          • modules
            • activation.cpp
            • batchnorm.cpp
            • conv.cpp
            • dropout.cpp
            • embedding.cpp
            • linear.cpp
            • loss.cpp
            • normalization.cpp
          • options
            • activation.cpp
            • dropout.cpp
            • normalization.cpp
        • serialize
          • input-archive.cpp
    • autograd
      • VariableTypeManual.cpp
      • VariableTypeUtils.h
      • autograd.cpp
      • custom_function.cpp
      • function.h
      • functions
        • accumulate_grad.cpp
        • tensor.cpp
        • utils.h
      • python_engine.cpp
      • python_function.cpp
      • python_legacy_variable.cpp
      • python_variable.cpp
      • saved_variable.cpp
      • variable.cpp
      • variable.h
    • cuda
      • comm.cpp
    • distributed
      • autograd
        • context
          • container.cpp
          • container.h
          • context.cpp
          • context.h
        • engine
          • dist_engine.cpp
          • dist_engine.h
        • functions
          • recvrpc_backward.h
        • init.cpp
        • utils.cpp
        • utils.h
      • c10d
        • reducer.cpp
      • rpc
        • init.cpp
        • python_functions.cpp
        • request_callback.cpp
        • request_callback_impl.cpp
    • jit
      • constants.cpp
      • export.cpp
      • graph_executor.cpp
      • import_legacy.cpp
      • import_source.cpp
      • init.cpp
      • instruction.cpp
      • ir.h
      • mobile
        • function.cpp
        • interpreter.cpp
        • register_mobile_ops.cpp
      • passes
        • python_print.cpp
        • quantization.cpp
      • profiling_graph_executor_impl.cpp
      • profiling_graph_executor_impl.h
      • pybind_utils.h
      • python_ir.cpp
      • python_tracer.cpp
      • register_prim_ops.cpp
      • register_special_ops.cpp
      • script
        • concrete_module_type.cpp
        • init.cpp
        • method.h
        • module.cpp
        • module.h
        • object.cpp
        • object.h
        • python_sugared_value.cpp
      • source_range.cpp
      • tracer.cpp
      • vararg_functions.cpp
      • vararg_functions.h
    • utils
      • pybind.h
      • tensor_numpy.cpp
      • throughput_benchmark.cpp
      • throughput_benchmark.h
  • distributed
    • autograd
      • __init__.py
    • rpc
      • __init__.py
      • api.py
      • backend_registry.py
  • jit
    • __init__.py
  • nn
    • intrinsic
      • __init__.py
      • modules
        • __init__.py
        • fused.py
      • quantized
        • __init__.py
        • modules
          • __init__.py
          • conv_relu.py
    • modules
      • fold.py
    • quantized/modules
      • __init__.py
      • conv.py
  • onnx
    • symbolic_opset11.py
  • quantization
    • default_mappings.py
  • serialization.py