Files

.ci
.circleci
.ctags.d
.devcontainer
.github
.vscode
android
aten
- conda
- src
  - ATen
    - benchmarks
    - core
    - cpu
    - cuda
    - cudnn
    - detail
    - functorch
    - hip
    - metal
    - miopen
    - mkl
    - mps
    - native
      - ao_sparse
      - cpu
      - cuda
      - cudnn
        AffineGridGenerator.cpp
        BatchNorm.cpp
        BatchNorm.h
        ConvPlaceholders.cpp
        ConvShared.cpp
        ConvShared.h
        Conv_v7.cpp
        Conv_v8.cpp
        GridSampler.cpp
        LossCTC.cpp
        MHA.cpp
        MHA.h
        RNN.cpp
        RNNUtils.h
      - hip
      - kleidiai
      - metal
      - miopen
      - mkl
      - mkldnn
      - mps
      - nested
      - quantized
      - sparse
      - transformers
      - ufunc
      - utils
      - vulkan
      - xnnpack
      - Activation.cpp
      - Activation.h
      - AdaptiveAveragePooling.cpp
      - AdaptiveAveragePooling3d.cpp
      - AdaptiveMaxPooling2d.cpp
      - AdaptiveMaxPooling3d.cpp
      - AdaptivePooling.h
      - AffineGridGenerator.cpp
      - AmpKernels.cpp
      - AmpKernels.h
      - AutogradComposite.cpp
      - AveragePool2d.cpp
      - AveragePool3d.cpp
      - BatchLinearAlgebra.cpp
      - BatchLinearAlgebra.h
      - BatchLinearAlgebraKernel.cpp
      - BinaryOps.cpp
      - BinaryOps.h
      - Blas.cpp
      - BlasKernel.cpp
      - Bucketization.cpp
      - BucketizationUtils.h
      - CPUBlas.cpp
      - CPUBlas.h
      - CPUFallback.cpp
      - CPUFallback.h
      - CanUse32BitIndexMath.h
      - ChanelShuffle.cpp
      - Col2Im.cpp
      - ComparisonUtils.cpp
      - ComplexHelper.h
      - CompositeRandomAccessor.h
      - CompositeRandomAccessorCommon.h
      - Constraints.cpp
      - ConvUtils.h
      - Convolution.cpp
      - ConvolutionMM2d.cpp
      - ConvolutionMM3d.cpp
      - ConvolutionMM3d.h
      - ConvolutionTBC.cpp
      - Copy.cpp
      - Copy.h
      - Correlation.cpp
      - 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
      - EmbeddingBag.h
      - Fill.cpp
      - Fill.h
      - ForeachOpsKernels.cpp
      - ForeachUtils.h
      - FractionalMaxPool2d.cpp
      - FractionalMaxPool3d.cpp
      - FractionalMaxPooling.h
      - FunctionOfAMatrixUtils.cpp
      - FunctionOfAMatrixUtils.h
      - FusedAdagrad.cpp
      - FusedAdagrad.h
      - FusedAdam.cpp
      - FusedAdam.h
      - FusedSGD.cpp
      - FusedSGD.h
      - GatedLinearUnit.cpp
      - Gelu.h
      - GridSampler.cpp
      - GridSampler.h
      - GridSamplerUtils.h
      - Histogram.cpp
      - Histogram.h
      - Im2Col.cpp
      - IndexKernel.h
      - IndexingUtils.cpp
      - IndexingUtils.h
      - Integration.cpp
      - Itertools.cpp
      - LegacyBatching.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
      - MathBitFallThroughLists.h
      - MathBitsFallback.h
      - MaxPooling.cpp
      - MaxPooling.h
      - MaxUnpooling.cpp
      - Memory.cpp
      - MetaTensor.cpp
      - NNPACK.cpp
      - NaiveConvolutionTranspose2d.cpp
      - NaiveConvolutionTranspose3d.cpp
      - NaiveDilatedConvolution.cpp
      - NamedTensor.cpp
      - NegateFallback.cpp
      - NonEmptyUtils.h
      - NonSymbolicBC.h
      - Normalization.cpp
      - Normalization.h
      - Onehot.cpp
      - PackedSequence.cpp
      - PadNd.cpp
      - Padding.h
      - PixelShuffle.cpp
      - PixelShuffle.h
      - PointwiseOps.cpp
      - PointwiseOps.h
      - Pool.h
      - Pooling.cpp
      - Pow.cpp
      - Pow.h
      - QuantizedLinear.cpp
      - README.md
      - RNN.cpp
      - RNN.h
      - RangeFactories.cpp
      - RangeFactories.h
      - RangeUtils.h
      - ReduceAllOps.cpp
      - ReduceAllOps.h
      - ReduceOps.cpp
      - ReduceOps.h
      - ReduceOpsUtils.h
      - ReductionType.h
      - ReflectionPad.cpp
      - Repeat.cpp
      - Repeat.h
      - ReplicationPadding.cpp
      - Resize.cpp
      - Resize.h
      - ResizeCommon.h
      - RowwisePrune.cpp
      - Scalar.cpp
      - ScatterGatherChecks.h
      - SegmentReduce.cpp
      - SegmentReduce.h
      - SharedReduceOps.h
      - SobolEngineOps.cpp
      - SobolEngineOpsUtils.cpp
      - SobolEngineOpsUtils.h
      - SoftMax.cpp
      - Sorting.cpp
      - Sorting.h
      - SortingUtils.h
      - SparseTensorUtils.cpp
      - SparseTensorUtils.h
      - SpectralOps.cpp
      - SpectralOpsUtils.h
      - StridedRandomAccessor.h
      - SummaryOps.cpp
      - TensorAdvancedIndexing.cpp
      - TensorAdvancedIndexing.h
      - TensorAdvancedIndexingUtils.h
      - TensorCompare.cpp
      - TensorCompare.h
      - TensorConversions.cpp
      - TensorConversions.h
      - TensorDimApply.h
      - TensorFactories.cpp
      - TensorFactories.h
      - TensorIterator.h
      - TensorIteratorDynamicCasting.h
      - TensorIteratorReduce.cpp
      - TensorProperties.cpp
      - TensorProperties.h
      - TensorShape.cpp
      - TensorShape.h
      - TensorTransformations.cpp
      - TensorTransformations.h
      - TestOps.cpp
      - TopKImpl.h
      - TransposeType.h
      - 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
      - prim_native_functions.cpp
      - tags.yaml
      - ts_native_functions.yaml
      - verbose_wrapper.cpp
      - verbose_wrapper.h
      - vol2col.h
    - nnapi
    - ops
    - quantized
    - templates
    - test
    - vulkan
    - xpu
    - .gitignore
    - ATen.h
    - ATenConfig.cmake.in
    - AccumulateType.cpp
    - AccumulateType.h
    - ArrayRef.h
    - Backend.h
    - Backtrace.h
    - BlasBackend.h
    - CMakeLists.txt
    - CPUApplyUtils.h
    - CPUFixedAllocator.h
    - CPUGeneratorImpl.cpp
    - CPUGeneratorImpl.h
    - CachedTensorUtils.cpp
    - CachedTensorUtils.h
    - CollapseDims.h
    - Config.h.in
    - ConjugateFallback.cpp
    - Context.cpp
    - Context.h
    - DLConvertor.cpp
    - DLConvertor.h
    - Device.h
    - DeviceAccelerator.cpp
    - DeviceAccelerator.h
    - DeviceGuard.h
    - DimVector.h
    - Dimname.h
    - Dispatch.cpp
    - Dispatch.h
    - Dispatch_v2.h
    - DynamicLibrary.cpp
    - DynamicLibrary.h
    - EmptyTensor.cpp
    - EmptyTensor.h
    - ExpandBase.h
    - ExpandUtils.cpp
    - ExpandUtils.h
    - Formatting.h
    - FuncTorchTLS.cpp
    - FuncTorchTLS.h
    - FunctionalInverses.cpp
    - FunctionalStorageImpl.cpp
    - FunctionalStorageImpl.h
    - FunctionalTensorWrapper.cpp
    - FunctionalTensorWrapper.h
    - FunctionalizeFallbackKernel.cpp
    - Generator.h
    - InferSize.h
    - InitialTensorOptions.h
    - Layout.h
    - LegacyBatchedFallback.cpp
    - LegacyBatchedFallback.h
    - LegacyBatchedTensorImpl.cpp
    - LegacyBatchedTensorImpl.h
    - LegacyBatchingRegistrations.cpp
    - LegacyVmapMode.cpp
    - LegacyVmapMode.h
    - LegacyVmapTransforms.cpp
    - LegacyVmapTransforms.h
    - LinalgBackend.h
    - MapAllocator.cpp
    - MapAllocator.h
    - MatrixRef.h
    - MemoryOverlap.cpp
    - MemoryOverlap.h
    - NamedTensor.h
    - NamedTensorUtils.cpp
    - NamedTensorUtils.h
    - NestedTensorImpl.cpp
    - NestedTensorImpl.h
    - NumericUtils.h
    - OpMathType.h
    - OpaqueTensorImpl.h
    - PTThreadPool.h
    - PadNd.h
    - Parallel-inl.h
    - Parallel.h
    - ParallelCommon.cpp
    - ParallelFuture.h
    - ParallelNative.cpp
    - ParallelNative.h
    - ParallelOpenMP.cpp
    - ParallelOpenMP.h
    - ParallelThreadPoolNative.cpp
    - PythonTorchFunctionTLS.cpp
    - PythonTorchFunctionTLS.h
    - ROCmFABackend.h
    - SDPBackend.h
    - SavedTensorHooks.cpp
    - SavedTensorHooks.h
    - Scalar.h
    - ScalarOps.cpp
    - ScalarOps.h
    - ScalarType.h
    - SequenceNumber.cpp
    - SequenceNumber.h
    - SmallVector.h
    - SparseCsrTensorImpl.cpp
    - SparseCsrTensorImpl.h
    - SparseCsrTensorUtils.h
    - SparseTensorImpl.cpp
    - SparseTensorImpl.h
    - Storage.h
    - StorageUtils.cpp
    - StorageUtils.h
    - Tensor.h
    - TensorAccessor.h
    - TensorGeometry.cpp
    - TensorGeometry.h
    - TensorIndexing.cpp
    - TensorIndexing.h
    - TensorIterator.cpp
    - TensorIterator.h
    - TensorIteratorInternal.h
    - TensorMeta.cpp
    - TensorMeta.h
    - TensorNames.cpp
    - TensorNames.h
    - TensorOperators.h
    - TensorOptions.h
    - TensorSubclassLikeUtils.h
    - TensorUtils.cpp
    - TensorUtils.h
    - ThreadLocalPythonObjects.cpp
    - ThreadLocalPythonObjects.h
    - ThreadLocalState.cpp
    - ThreadLocalState.h
    - TracerMode.h
    - TypeDefault.h
    - Utils.cpp
    - Utils.h
    - Version.cpp
    - Version.h
    - VmapModeRegistrations.cpp
    - WrapDimUtils.h
    - WrapDimUtilsMulti.h
    - ZeroTensorFallback.cpp
    - autocast_mode.cpp
    - autocast_mode.h
    - ceil_div.h
    - code_template.h
    - cpp_custom_type_hack.h
    - div_rtn.h
    - dlpack.h
    - jit_macros.h
    - jiterator_macros.h
    - record_function.cpp
    - record_function.h
  - THC
  - README.md
- tools
- CMakeLists.txt
benchmarks
binaries
c10
caffe2
cmake
docs
functorch
mypy_plugins
scripts
test
third_party
tools
torch
torchgen
.bazelignore
.bazelrc
.bazelversion
.clang-format
.clang-tidy
.cmakelintrc
.coveragerc
.dockerignore
.flake8
.gdbinit
.git-blame-ignore-revs
.gitattributes
.gitignore
.gitmodules
.lintrunner.toml
.lldbinit
BUCK.oss
BUILD.bazel
CITATION.cff
CMakeLists.txt
CODEOWNERS
CODE_OF_CONDUCT.md
CONTRIBUTING.md
Dockerfile
GLOSSARY.md
LICENSE
MANIFEST.in
Makefile
NOTICE
README.md
RELEASE.md
SECURITY.md
WORKSPACE
aten.bzl
buckbuild.bzl
build.bzl
build_variables.bzl
defs.bzl
docker.Makefile
mypy-strict.ini
mypy.ini
pt_ops.bzl
pt_template_srcs.bzl
pyproject.toml
pytest.ini
requirements.txt
setup.py
ubsan.supp
ufunc_defs.bzl
version.txt

BatchNorm.h

and

pytorchmergebot

Batch Norm Consolidation (pytorch#116092 )

Mar 18, 2024

773ae81 · Mar 18, 2024