Files

.circleci
.ctags.d
.github
.jenkins
android
aten
- conda
- src
  - ATen
    - benchmarks
    - core
    - cpu
    - cuda
    - cudnn
    - detail
    - hip
    - metal
    - miopen
    - mkl
    - native
      - cpu
      - cuda
      - cudnn
      - metal
      - miopen
      - mkl
        LinearAlgebra.cpp
        SpectralOps.cpp
      - mkldnn
      - quantized
      - sparse
      - utils
      - vulkan
      - xnnpack
      - Activation.cpp
      - Activation.h
      - AdaptiveAveragePooling.cpp
      - AdaptiveAveragePooling3d.cpp
      - AdaptiveMaxPooling2d.cpp
      - AdaptiveMaxPooling3d.cpp
      - AdaptivePooling.h
      - AffineGridGenerator.cpp
      - AutogradComposite.cpp
      - AveragePool2d.cpp
      - AveragePool3d.cpp
      - BatchLinearAlgebra.cpp
      - BatchLinearAlgebra.h
      - BatchLinearAlgebraKernel.cpp
      - Batching.cpp
      - BinaryOps.cpp
      - BinaryOps.h
      - Blas.cpp
      - BlasKernel.cpp
      - Bucketization.cpp
      - BucketizationUtils.h
      - CPUBlas.cpp
      - CPUBlas.h
      - ChanelShuffle.cpp
      - Col2Im.cpp
      - ComplexHelper.h
      - CompositeRandomAccessor.h
      - CompositeRandomAccessorCommon.h
      - ConstantPadNd.cpp
      - ConvUtils.h
      - Convolution.cpp
      - ConvolutionMM2d.cpp
      - ConvolutionMM3d.cpp
      - ConvolutionTBC.cpp
      - Copy.cpp
      - Copy.h
      - Cross.cpp
      - Cross.h
      - DilatedConvolutionUtils.h
      - DilatedMaxPool2d.cpp
      - DilatedMaxPool3d.cpp
      - DispatchStub.cpp
      - DispatchStub.h
      - Distance.cpp
      - Distance.h
      - DistributionTemplates.h
      - Distributions.cpp
      - Distributions.h
      - Dropout.cpp
      - Embedding.cpp
      - EmbeddingBag.cpp
      - Fill.cpp
      - Fill.h
      - ForeachOpsKernels.cpp
      - ForeachUtils.h
      - FractionalMaxPool2d.cpp
      - FractionalMaxPool3d.cpp
      - FunctionOfAMatrixUtils.cpp
      - FunctionOfAMatrixUtils.h
      - GatedLinearUnit.cpp
      - GridSampler.cpp
      - GridSampler.h
      - Im2Col.cpp
      - IndexingUtils.cpp
      - IndexingUtils.h
      - Integration.cpp
      - Itertools.cpp
      - LegacyBridge.cpp
      - LegacyNNDefinitions.cpp
      - Lerp.cpp
      - Lerp.h
      - Linear.cpp
      - LinearAlgebra.cpp
      - LinearAlgebra.h
      - LinearAlgebraUtils.h
      - Loss.cpp
      - LossCTC.cpp
      - LossMulti.h
      - LossMultiLabelMargin.cpp
      - LossMultiMargin.cpp
      - LossNLL.cpp
      - LossNLL2d.cpp
      - Math.h
      - MaxPooling.cpp
      - MaxPooling.h
      - MaxUnpooling.cpp
      - Memory.cpp
      - MetaTensor.cpp
      - NNPACK.cpp
      - NaiveConvolutionTranspose2d.cpp
      - NaiveConvolutionTranspose3d.cpp
      - NaiveDilatedConvolution.cpp
      - NamedTensor.cpp
      - Normalization.cpp
      - Onehot.cpp
      - PackedSequence.cpp
      - PixelShuffle.cpp
      - PointwiseOps.cpp
      - PointwiseOps.h
      - Pool.h
      - Pooling.cpp
      - Pow.cpp
      - Pow.h
      - QuantizedLinear.cpp
      - README.md
      - RNN.cpp
      - RNN.h
      - RangeFactories.cpp
      - ReduceAllOps.cpp
      - ReduceAllOps.h
      - ReduceOps.cpp
      - ReduceOps.h
      - ReduceOpsUtils.h
      - ReflectionPad.cpp
      - Repeat.cpp
      - Repeat.h
      - ReplicationPadding.cpp
      - Resize.cpp
      - Resize.h
      - ResizeCommon.h
      - RowwisePrune.cpp
      - Scalar.cpp
      - ScatterGatherChecks.h
      - SharedReduceOps.h
      - SobolEngineOps.cpp
      - SobolEngineOpsUtils.cpp
      - SobolEngineOpsUtils.h
      - SoftMax.cpp
      - Sorting.cpp
      - Sorting.h
      - SortingUtils.h
      - SpectralOps.cpp
      - SpectralOpsUtils.h
      - StridedRandomAccessor.h
      - SummaryOps.cpp
      - TensorAdvancedIndexing.cpp
      - TensorAdvancedIndexing.h
      - TensorCompare.cpp
      - TensorCompare.h
      - TensorConversions.cpp
      - TensorDimApply.h
      - TensorFactories.cpp
      - TensorFactories.h
      - TensorIterator.h
      - TensorIteratorDynamicCasting.h
      - TensorIteratorReduce.cpp
      - TensorProperties.cpp
      - TensorShape.cpp
      - TensorTransformations.cpp
      - TensorTransformations.h
      - TestOps.cpp
      - TriangularOps.cpp
      - TriangularOpsUtils.h
      - TypeProperties.cpp
      - TypeProperties.h
      - UnaryOps.cpp
      - UnaryOps.h
      - Unfold2d.cpp
      - Unfold2d.h
      - Unfold3d.cpp
      - Unfold3d.h
      - UnfoldBackward.cpp
      - UnfoldBackward.h
      - Unique.cpp
      - UpSample.cpp
      - UpSample.h
      - UpSampleBicubic2d.cpp
      - UpSampleBilinear2d.cpp
      - UpSampleLinear1d.cpp
      - UpSampleNearest1d.cpp
      - UpSampleNearest2d.cpp
      - UpSampleNearest3d.cpp
      - UpSampleTrilinear3d.cpp
      - VariableMethodStubs.cpp
      - WeightNorm.cpp
      - batch_norm.h
      - group_norm.cpp
      - group_norm.h
      - im2col.h
      - im2col_shape_check.h
      - layer_norm.cpp
      - layer_norm.h
      - native_functions.yaml
      - vol2col.h
    - nnapi
    - quantized
    - templates
    - test
    - vulkan
    - .gitignore
    - ATen.h
    - ATenConfig.cmake.in
    - AccumulateType.h
    - ArrayRef.h
    - Backend.h
    - Backtrace.h
    - BatchedFallback.cpp
    - BatchedFallback.h
    - BatchedTensorImpl.cpp
    - BatchedTensorImpl.h
    - BatchingRegistrations.cpp
    - CMakeLists.txt
    - CPUApplyUtils.h
    - CPUFixedAllocator.h
    - CPUGeneratorImpl.cpp
    - CPUGeneratorImpl.h
    - CUDAGeneratorImpl.h
    - Config.h.in
    - Context.cpp
    - Context.h
    - DLConvertor.cpp
    - DLConvertor.h
    - Device.h
    - DeviceGuard.h
    - DimVector.h
    - Dimname.h
    - Dispatch.h
    - DynamicLibrary.cpp
    - DynamicLibrary.h
    - ExpandUtils.cpp
    - ExpandUtils.h
    - Formatting.h
    - Generator.h
    - InferSize.h
    - InitialTensorOptions.h
    - Layout.h
    - LegacyTHFunctionsCPU.cpp
    - LegacyTHFunctionsCPU.h
    - LegacyTHFunctionsCUDA.h
    - MatrixRef.h
    - MemoryOverlap.cpp
    - MemoryOverlap.h
    - NamedTensor.h
    - NamedTensorUtils.cpp
    - NamedTensorUtils.h
    - NumericUtils.h
    - OpaqueTensorImpl.h
    - PTThreadPool.h
    - Parallel.h
    - ParallelCommon.cpp
    - ParallelNative.cpp
    - ParallelNative.h
    - ParallelNativeTBB.cpp
    - ParallelNativeTBB.h
    - ParallelOpenMP.cpp
    - ParallelOpenMP.h
    - ParallelThreadPoolNative.cpp
    - Scalar.h
    - ScalarOps.cpp
    - ScalarOps.h
    - ScalarType.h
    - SequenceNumber.cpp
    - SequenceNumber.h
    - SmallVector.h
    - SparseTensorImpl.cpp
    - SparseTensorImpl.h
    - SparseTensorUtils.cpp
    - SparseTensorUtils.h
    - Storage.h
    - Tensor.h
    - TensorAccessor.h
    - TensorGeometry.cpp
    - TensorGeometry.h
    - TensorIndexing.cpp
    - TensorIndexing.h
    - TensorIterator.cpp
    - TensorIterator.h
    - TensorMeta.cpp
    - TensorMeta.h
    - TensorNames.cpp
    - TensorNames.h
    - TensorOperators.h
    - TensorOptions.h
    - TensorUtils.cpp
    - TensorUtils.h
    - ThreadLocalState.cpp
    - ThreadLocalState.h
    - TracerMode.h
    - TypeDefault.h
    - Utils.cpp
    - Utils.h
    - Version.cpp
    - Version.h
    - VmapMode.cpp
    - VmapMode.h
    - VmapModeRegistrations.cpp
    - VmapTransforms.cpp
    - VmapTransforms.h
    - WrapDimUtils.h
    - WrapDimUtilsMulti.h
    - autocast_mode.cpp
    - autocast_mode.h
    - cpp_custom_type_hack.h
    - div_rtn.h
    - dlpack.h
    - function_wrapper.py
    - gen_vulkan_glsl.py
    - gen_vulkan_spv.py
    - record_function.cpp
    - record_function.h
  - TH
  - THC
  - THCUNN
  - README.md
- tools
- CMakeLists.txt
benchmarks
binaries
c10
caffe2
cmake
docker
docs
ios
modules
mypy_plugins
scripts
submodules
test
third_party
tools
torch
.bazelrc
.bazelversion
.clang-format
.clang-tidy
.cmakelintrc
.dockerignore
.flake8
.gitattributes
.gitignore
.gitmodules
.nojekyll
BUILD.bazel
CITATION
CMakeLists.txt
CODEOWNERS
CODE_OF_CONDUCT.md
CONTRIBUTING.md
Dockerfile
GLOSSARY.md
LICENSE
Makefile
NOTICE
README.md
WORKSPACE
aten.bzl
codecov.yml
docker.Makefile
mypy-strict.ini
mypy.ini
requirements-flake8.txt
requirements.txt
setup.py
ubsan.supp
version.txt