|
Go up
|
— |
— |
|
|
ao_sparse/
|
— |
|
|
|
cpu/
|
— |
|
|
|
cuda/
|
— |
|
|
|
hip/
|
— |
|
|
|
kleidiai/
|
— |
|
|
|
mps/
|
— |
|
|
|
nested/
|
— |
|
|
|
quantized/
|
— |
|
|
|
transformers/
|
— |
|
|
|
utils/
|
— |
|
|
|
Activation.h
|
3.7 KiB |
|
|
|
AdaptivePooling.h
|
2.6 KiB |
|
|
|
AmpKernels.h
|
871 B |
|
|
|
batch_norm.h
|
1.6 KiB |
|
|
|
BatchLinearAlgebra.h
|
10 KiB |
|
|
|
BinaryOps.h
|
6.2 KiB |
|
|
|
BucketizationUtils.h
|
7.8 KiB |
|
|
|
CanUse32BitIndexMath.h
|
496 B |
|
|
|
ComplexHelper.h
|
4.5 KiB |
|
|
|
CompositeRandomAccessor.h
|
1.1 KiB |
|
|
|
CompositeRandomAccessorCommon.h
|
6.8 KiB |
|
|
|
ConvolutionMM3d.h
|
594 B |
|
|
|
ConvUtils.h
|
21 KiB |
|
|
|
Copy.h
|
626 B |
|
|
|
CPUBlas.h
|
9.5 KiB |
|
|
|
CPUFallback.h
|
2.6 KiB |
|
|
|
Cross.h
|
513 B |
|
|
|
DilatedConvolutionUtils.h
|
6.5 KiB |
|
|
|
DispatchStub.h
|
16 KiB |
|
|
|
Distance.h
|
974 B |
|
|
|
Distributions.h
|
21 KiB |
|
|
|
DistributionTemplates.h
|
18 KiB |
|
|
|
EmbeddingBag.h
|
5.4 KiB |
|
|
|
Fill.h
|
650 B |
|
|
|
ForeachUtils.h
|
15 KiB |
|
|
|
FractionalMaxPooling.h
|
2.4 KiB |
|
|
|
FunctionOfAMatrixUtils.h
|
642 B |
|
|
|
FusedAdagrad.h
|
749 B |
|
|
|
FusedAdam.h
|
937 B |
|
|
|
FusedSGD.h
|
770 B |
|
|
|
Gelu.h
|
1.1 KiB |
|
|
|
GridSampler.h
|
10 KiB |
|
|
|
GridSamplerUtils.h
|
3.9 KiB |
|
|
|
group_norm.h
|
1.1 KiB |
|
|
|
GroupedMMUtils.h
|
7.4 KiB |
|
|
|
Histogram.h
|
999 B |
|
|
|
im2col.h
|
5.4 KiB |
|
|
|
im2col_shape_check.h
|
7.4 KiB |
|
|
|
IndexingUtils.h
|
6.4 KiB |
|
|
|
IndexKernel.h
|
1.9 KiB |
|
|
|
layer_norm.h
|
4.8 KiB |
|
|
|
Lerp.h
|
1.7 KiB |
|
|
|
LinearAlgebra.h
|
553 B |
|
|
|
LinearAlgebraUtils.h
|
26 KiB |
|
|
|
LossMulti.h
|
2.5 KiB |
|
|
|
Math.h
|
139 KiB |
|
|
|
MathBitFallThroughLists.h
|
4.3 KiB |
|
|
|
MathBitsFallback.h
|
7.4 KiB |
|
|
|
MaxPooling.h
|
3.4 KiB |
|
|
|
NonEmptyUtils.h
|
853 B |
|
|
|
NonSymbolicBC.h
|
3.1 KiB |
|
|
|
Normalization.h
|
807 B |
|
|
|
Padding.h
|
2.3 KiB |
|
|
|
PixelShuffle.h
|
2.2 KiB |
|
|
|
PointwiseOps.h
|
1.0 KiB |
|
|
|
Pool.h
|
13 KiB |
|
|
|
Pow.h
|
1.9 KiB |
|
|
|
RangeFactories.h
|
608 B |
|
|
|
RangeUtils.h
|
2.3 KiB |
|
|
|
ReduceAllOps.h
|
651 B |
|
|
|
ReduceOps.h
|
2.1 KiB |
|
|
|
ReduceOpsUtils.h
|
16 KiB |
|
|
|
ReductionType.h
|
1.4 KiB |
|
|
|
Repeat.h
|
1.7 KiB |
|
|
|
Resize.h
|
9.6 KiB |
|
|
|
ResizeCommon.h
|
2.7 KiB |
|
|
|
RNN.h
|
2.7 KiB |
|
|
|
ScaledBlasUtils.h
|
3.5 KiB |
|
|
|
ScatterGatherChecks.h
|
3.9 KiB |
|
|
|
SegmentReduce.h
|
1.5 KiB |
|
|
|
SharedReduceOps.h
|
16 KiB |
|
|
|
SobolEngineOpsUtils.h
|
2.0 KiB |
|
|
|
Sorting.h
|
870 B |
|
|
|
SortingUtils.h
|
2.9 KiB |
|
|
|
SparseTensorUtils.h
|
6.6 KiB |
|
|
|
SpectralOpsUtils.h
|
3.5 KiB |
|
|
|
StridedRandomAccessor.h
|
6.9 KiB |
|
|
|
TensorAdvancedIndexing.h
|
3.1 KiB |
|
|
|
TensorAdvancedIndexingUtils.h
|
3.3 KiB |
|
|
|
TensorCompare.h
|
1.7 KiB |
|
|
|
TensorConversions.h
|
1.1 KiB |
|
|
|
TensorDimApply.h
|
2.0 KiB |
|
|
|
TensorFactories.h
|
5.5 KiB |
|
|
|
TensorIterator.h
|
300 B |
|
|
|
TensorIteratorDynamicCasting.h
|
2.0 KiB |
|
|
|
TensorProperties.h
|
452 B |
|
|
|
TensorShape.h
|
4.7 KiB |
|
|
|
TensorTransformations.h
|
1.2 KiB |
|
|
|
TopKImpl.h
|
3.6 KiB |
|
|
|
TransposeType.h
|
935 B |
|
|
|
TriangularOpsUtils.h
|
2.2 KiB |
|
|
|
TypeProperties.h
|
912 B |
|
|
|
UnaryOps.h
|
5.5 KiB |
|
|
|
Unfold2d.h
|
1.2 KiB |
|
|
|
Unfold3d.h
|
1.1 KiB |
|
|
|
UnfoldBackward.h
|
3.3 KiB |
|
|
|
UpSample.h
|
19 KiB |
|
|
|
verbose_wrapper.h
|
447 B |
|
|
|
vol2col.h
|
3.7 KiB |
|
|