cpu

cuda

mps

nested

quantized

transformers

utils

Activation.h

AdaptivePooling.h

AmpKernels.h

BatchLinearAlgebra.h

BinaryOps.h

BucketizationUtils.h

CPUBlas.h

CPUFallback.h

CanUse32BitIndexMath.h

ComplexHelper.h

CompositeRandomAccessor.h

CompositeRandomAccessorCommon.h

ConvUtils.h

ConvolutionMM3d.h

Copy.h

Cross.h

DilatedConvolutionUtils.h

DispatchStub.h

Distance.h

DistributionTemplates.h

Distributions.h

EmbeddingBag.h

Fill.h

ForeachUtils.h

FractionalMaxPooling.h

FunctionOfAMatrixUtils.h

FusedAdagrad.h

FusedAdam.h

FusedSGD.h

GridSampler.h

GridSamplerUtils.h

Histogram.h

IndexKernel.h

IndexingUtils.h

Lerp.h

LinearAlgebra.h

LinearAlgebraUtils.h

LossMulti.h

Math.h

MathBitFallThroughLists.h

MathBitsFallback.h

MaxPooling.h

NonEmptyUtils.h

NonSymbolicBC.h

Normalization.h

Padding.h

PixelShuffle.h

PointwiseOps.h

Pool.h

Pow.h

RNN.h

RangeFactories.h

ReduceAllOps.h

ReduceOps.h

ReduceOpsUtils.h

ReductionType.h

Repeat.h

Resize.h

ResizeCommon.h

ScatterGatherChecks.h

SegmentReduce.h

SharedReduceOps.h

SobolEngineOpsUtils.h

Sorting.h

SortingUtils.h

SparseTensorUtils.h

SpectralOpsUtils.h

StridedRandomAccessor.h

TensorAdvancedIndexing.h

TensorAdvancedIndexingUtils.h

TensorCompare.h

TensorConversions.h

TensorDimApply.h

TensorFactories.h

TensorIterator.h

TensorIteratorDynamicCasting.h

TensorProperties.h

TensorShape.h

TensorTransformations.h

TopKImpl.h

TransposeType.h

TriangularOpsUtils.h

TypeProperties.h

UnaryOps.h

Unfold2d.h

Unfold3d.h

UnfoldBackward.h

UpSample.h

batch_norm.h

group_norm.h

im2col.h

im2col_shape_check.h

layer_norm.h

verbose_wrapper.h

vol2col.h