Directory listing for /PyTorch/pytorch/aten/src/ATen/native/
Activation.cpp
Activation.h
AdaptiveAveragePooling.cpp
AdaptiveAveragePooling3d.cpp
AdaptiveMaxPooling2d.cpp
AdaptiveMaxPooling3d.cpp
AdaptivePooling.h
AffineGridGenerator.cpp
AmpKernels.cpp
AmpKernels.h
ao_sparse/
AutogradComposite.cpp
AveragePool2d.cpp
AveragePool3d.cpp
batch_norm.h
BatchLinearAlgebra.cpp
BatchLinearAlgebra.h
BatchLinearAlgebraKernel.cpp
BinaryOps.cpp
BinaryOps.h
Blas.cpp
BlasKernel.cpp
Bucketization.cpp
BucketizationUtils.h
CanUse32BitIndexMath.h
ChanelShuffle.cpp
Col2Im.cpp
ComparisonUtils.cpp
ComplexHelper.h
CompositeRandomAccessor.h
CompositeRandomAccessorCommon.h
Constraints.cpp
Convolution.cpp
ConvolutionMM2d.cpp
ConvolutionMM3d.cpp
ConvolutionMM3d.h
ConvolutionTBC.cpp
ConvUtils.h
Copy.cpp
Copy.h
Correlation.cpp
cpu/
CPUBlas.cpp
CPUBlas.h
CPUFallback.cpp
CPUFallback.h
Cross.cpp
Cross.h
cuda/
cudnn/
DilatedConvolutionUtils.h
DilatedMaxPool2d.cpp
DilatedMaxPool3d.cpp
DispatchStub.cpp
DispatchStub.h
Distance.cpp
Distance.h
Distributions.cpp
Distributions.h
DistributionTemplates.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
group_norm.cpp
group_norm.h
hip/
Histogram.cpp
Histogram.h
Im2Col.cpp
im2col.h
im2col_shape_check.h
IndexingUtils.cpp
IndexingUtils.h
IndexKernel.h
Integration.cpp
Itertools.cpp
kleidiai/
layer_norm.cpp
layer_norm.h
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
metal/
MetaTensor.cpp
miopen/
mkl/
mkldnn/
mps/
mtia/
NaiveConvolutionTranspose2d.cpp
NaiveConvolutionTranspose3d.cpp
NaiveDilatedConvolution.cpp
NamedTensor.cpp
native_functions.yaml
NegateFallback.cpp
nested/
NNPACK.cpp
NonEmptyUtils.h
NonSymbolicBC.h
Normalization.cpp
Normalization.h
Onehot.cpp
PackedSequence.cpp
Padding.h
PadNd.cpp
PixelShuffle.cpp
PixelShuffle.h
PointwiseOps.cpp
PointwiseOps.h
Pool.h
Pooling.cpp
Pow.cpp
Pow.h
prim_native_functions.cpp
quantized/
QuantizedLinear.cpp
RangeFactories.cpp
RangeFactories.h
RangeUtils.h
README.md
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
RNN.cpp
RNN.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
sparse/
SparseTensorUtils.cpp
SparseTensorUtils.h
SpectralOps.cpp
SpectralOpsUtils.h
StridedRandomAccessor.h
SummaryOps.cpp
tags.yaml
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
transformers/
TransposeType.h
TriangularOps.cpp
TriangularOpsUtils.h
ts_native_functions.yaml
TypeProperties.cpp
TypeProperties.h
ufunc/
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
utils/
VariableMethodStubs.cpp
verbose_wrapper.cpp
verbose_wrapper.h
vol2col.h
vulkan/
WeightNorm.cpp
xnnpack/