pytorch/pytorch/aten/src/ATen/native/ Source Tree

../
Activation.cpp
Activation.h GeluType
AdaptiveAveragePooling.cpp
AdaptiveAveragePooling3d.cpp
AdaptiveMaxPooling2d.cpp
AdaptiveMaxPooling3d.cpp
AdaptivePooling.h
AffineGridGenerator.cpp
AutogradComposite.cpp
AveragePool2d.cpp
AveragePool3d.cpp
BatchLinearAlgebra.cpp
BatchLinearAlgebra.h LapackLstsqDriverType
BatchLinearAlgebraKernel.cpp
BinaryOps.cpp
BinaryOps.h
Blas.cpp
BlasKernel.cpp
Bucketization.cpp
BucketizationUtils.h
CPUBlas.cpp
CPUBlas.h
CPUFallback.cpp
CPUFallback.h _call_fallback_fn
CanUse32BitIndexMath.h
ChanelShuffle.cpp
Col2Im.cpp
ComparisonUtils.cpp
ComplexHelper.h
CompositeRandomAccessor.h TupleInfoCPU
CompositeRandomAccessorCommon.h at::native::(anonymous namespace)::operator_brackets_proxy CompositeRandomAccessor references_holder
ConvUtils.h ConvBackend
Convolution.cpp ConvParams
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 at::native::(anonymous namespace)::RegisterCUDADispatch at::native::(anonymous namespace)::RegisterHIPDispatch at::native::(anonymous namespace)::RegisterMPSDispatch CPUCapability DispatchStub DispatchStubImpl
Distance.cpp
Distance.h
DistributionTemplates.h
Distributions.cpp BernoulliStub CauchyStub ExponentialStub GeometricStub LogNormalStub NormalMeta NormalStub RandomFromToMeta RandomFromToStub RandomStub UniformMeta UniformStub
Distributions.h (anonymous namespace)::BaseSampler
Dropout.cpp
Embedding.cpp
EmbeddingBag.cpp
EmbeddingBag.h _CallbackAndBlockSize _EmbeddingBagKernelCacheImpl
Fill.cpp
Fill.h
ForeachOpsKernels.cpp
ForeachUtils.h
FractionalMaxPool2d.cpp
FractionalMaxPool3d.cpp
FractionalMaxPooling.h
FunctionOfAMatrixUtils.cpp
FunctionOfAMatrixUtils.h
GatedLinearUnit.cpp
GridSampler.cpp
GridSampler.h
GridSamplerUtils.h GridSamplerInterpolation GridSamplerPadding
Histogram.cpp
Histogram.h
Im2Col.cpp
IndexKernel.h
IndexingUtils.cpp
IndexingUtils.h AdvancedIndex
Integration.cpp
Itertools.cpp
LegacyBatching.cpp
LegacyBridge.cpp
Lerp.cpp
Lerp.h
Linear.cpp
LinearAlgebra.cpp
LinearAlgebra.h
LinearAlgebraUtils.h BroadcastLinearIndices
Loss.cpp
LossCTC.cpp
LossMulti.h
LossMultiLabelMargin.cpp
LossMultiMargin.cpp
LossNLL.cpp
LossNLL2d.cpp
Math.h
MathBitFallThroughLists.h
MathBitsFallback.h MathOpFallback
MaxPooling.cpp
MaxPooling.h PoolingParams1D
MaxUnpooling.cpp
Memory.cpp
MetaTensor.cpp
NNPACK.cpp
NaiveConvolutionTranspose2d.cpp
NaiveConvolutionTranspose3d.cpp
NaiveDilatedConvolution.cpp
NamedTensor.cpp
NegateFallback.cpp NegFallback
NonEmptyUtils.h
NonSymbolicBC.h
Normalization.cpp InvStd Var
Normalization.h
Onehot.cpp
PackedSequence.cpp
PadNd.cpp
PixelShuffle.cpp
PointwiseOps.cpp
PointwiseOps.h
Pool.h
Pooling.cpp
Pow.cpp
Pow.h
QuantizedLinear.cpp
RNN.cpp
RNN.h
RangeFactories.cpp
RangeFactories.h
ReduceAllOps.cpp
ReduceAllOps.h
ReduceOps.cpp
ReduceOps.h
ReduceOpsUtils.h
ReductionType.h ReductionType
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 AbsMaxOps AbsMinOps AbsSwitch ArgMaxOps ArgMinOps MaxOps MeanOps MinMaxOps MinOps NanSumOps NormOneOps NormOps NormTwoOps NormZeroOps WelfordData WelfordOps ArgReductionOps GreaterOrNan LessOrNan MinMaxReductionOps
SobolEngineOps.cpp
SobolEngineOpsUtils.cpp
SobolEngineOpsUtils.h
SoftMax.cpp
Sorting.cpp
Sorting.h QUANTILE_INTERPOLATION_MODE
SortingUtils.h
SpectralOps.cpp
SpectralOpsUtils.h fft_norm_mode
StridedRandomAccessor.h ConstStridedRandomAccessor DefaultPtrTraits RestrictPtrTraits StridedRandomAccessor
SummaryOps.cpp
TensorAdvancedIndexing.cpp
TensorAdvancedIndexing.h
TensorAdvancedIndexingUtils.h
TensorCompare.cpp
TensorCompare.h ClampLimits
TensorConversions.cpp
TensorConversions.h
TensorDimApply.h
TensorFactories.cpp
TensorFactories.h ZeroTensorAllocator
TensorIterator.h
TensorIteratorDynamicCasting.h needs_dynamic_casting
TensorIteratorReduce.cpp
TensorProperties.cpp
TensorProperties.h
TensorShape.cpp
TensorShape.h
TensorTransformations.cpp
TensorTransformations.h
TestOps.cpp
TopKImpl.h
TransposeType.h TransposeType
TriangularOps.cpp
TriangularOpsUtils.h
TypeProperties.cpp
TypeProperties.h ResultTypeState
UnaryOps.cpp
UnaryOps.h
Unfold2d.cpp
Unfold2d.h
Unfold3d.cpp
Unfold3d.h
UnfoldBackward.cpp
UnfoldBackward.h
Unique.cpp hash
UpSample.cpp
UpSample.h
UpSampleBicubic2d.cpp
UpSampleBilinear2d.cpp
UpSampleLinear1d.cpp
UpSampleNearest1d.cpp
UpSampleNearest2d.cpp
UpSampleNearest3d.cpp
UpSampleTrilinear3d.cpp
VariableMethodStubs.cpp
WeightNorm.cpp
[+] ao_sparse/
batch_norm.h
[+] cpu/
[+] cuda/
[+] cudnn/
group_norm.cpp
group_norm.h
im2col.h
im2col_shape_check.h
layer_norm.cpp
layer_norm.h
[+] miopen/
[+] mkl/
[+] mkldnn/
[+] mps/
[+] nested/
prim_native_functions.cpp
[+] quantized/
[+] sparse/
[+] transformers/
[+] ufunc/
[+] utils/
verbose_wrapper.cpp
verbose_wrapper.h
vol2col.h
[+] vulkan/
[+] xnnpack/

Browse the source code of pytorch/pytorch/aten/src/ATen/native/ online