| Parent directory/ | - | - |
| Activation.cpp | 59.1 KiB | 2024-Oct-31 02:38 |
| AdaptiveAvgPoolKernel.cpp | 31.8 KiB | 2024-Nov-02 04:59 |
| AdaptiveMaxPoolKernel.cpp | 38.7 KiB | 2024-Nov-02 04:59 |
| AmpGradScalerKernels.cpp | 8.1 KiB | 2024-Nov-02 04:59 |
| AtomicAddFloat.h | 857 B | 2024-Oct-31 02:38 |
| AvgPoolKernel.cpp | 40.3 KiB | 2024-Nov-02 04:59 |
| BinaryOpsKernel.cpp | 53.0 KiB | 2024-Nov-02 04:59 |
| BlasKernel.cpp | 15.9 KiB | 2024-Nov-02 04:59 |
| CatKernel.cpp | 2.2 KiB | 2024-Oct-31 02:38 |
| CatKernel.h | 307 B | 2024-Nov-02 04:59 |
| ChannelShuffleKernel.cpp | 3.7 KiB | 2024-Oct-31 02:38 |
| ChannelShuffleKernel.h | 287 B | 2024-Nov-02 04:59 |
| ComplexKernel.cpp | 925 B | 2024-Oct-31 02:38 |
| CopyKernel.cpp | 13.3 KiB | 2024-Oct-31 02:38 |
| CopyKernel.h | 312 B | 2024-Nov-02 04:59 |
| CrossKernel.cpp | 2.8 KiB | 2024-Oct-31 02:38 |
| DepthwiseConvKernel.cpp | 18.6 KiB | 2024-Nov-02 04:59 |
| DepthwiseConvKernel.h | 471 B | 2024-Oct-31 02:38 |
| DistanceOpsKernel.cpp | 18.4 KiB | 2024-Oct-31 02:38 |
| DistributionKernels.cpp | 10.6 KiB | 2024-Oct-31 02:38 |
| DistributionTemplates.h | 16.1 KiB | 2024-Nov-02 04:59 |
| FillKernel.cpp | 2.7 KiB | 2024-Nov-02 04:59 |
| FlashAttentionKernel.cpp | 31.6 KiB | 2024-Oct-31 02:38 |
| FunctionOfAMatrixUtilsKernel.cpp | 1.6 KiB | 2024-Oct-31 02:38 |
| FusedAdagradKernel.cpp | 7.0 KiB | 2024-Nov-02 04:59 |
| FusedAdamKernel.cpp | 14.2 KiB | 2024-Nov-02 04:59 |
| FusedSGDKernel.cpp | 8.5 KiB | 2024-Nov-02 04:59 |
| GridSamplerKernel.cpp | 56.2 KiB | 2024-Nov-02 04:59 |
| GridSamplerKernel.h | 825 B | 2024-Nov-02 04:59 |
| HistogramKernel.cpp | 13.1 KiB | 2024-Nov-02 04:59 |
| IndexKernel.cpp | 30.4 KiB | 2024-Nov-02 04:59 |
| IndexKernelUtils.h | 2.9 KiB | 2024-Nov-02 04:59 |
| Intrinsics.h | 1.2 KiB | 2024-Oct-31 02:38 |
| IsContiguous.h | 2.4 KiB | 2024-Nov-02 04:59 |
| LerpKernel.cpp | 6.5 KiB | 2024-Oct-31 02:38 |
| LinearAlgebraKernel.cpp | 2.7 KiB | 2024-Oct-31 02:38 |
| LogAddExp.h | 2.4 KiB | 2024-Nov-02 04:59 |
| Loops.h | 14.5 KiB | 2024-Nov-02 04:59 |
| MaxPoolKernel.cpp | 27.7 KiB | 2024-Nov-02 04:59 |
| MaxPooling.cpp | 1.8 KiB | 2024-Oct-31 02:38 |
| MaxUnpoolKernel.cpp | 8.8 KiB | 2024-Nov-02 04:59 |
| MaxUnpoolKernel.h | 308 B | 2024-Oct-31 02:38 |
| MultinomialKernel.cpp | 8.7 KiB | 2024-Oct-31 02:38 |
| NativeMultiheadAttnKernel.cpp | 3.6 KiB | 2024-Oct-31 02:38 |
| PaddingKernel.cpp | 27.8 KiB | 2024-Oct-31 02:38 |
| PixelShuffleKernel.cpp | 9.0 KiB | 2024-Oct-31 02:38 |
| PixelShuffleKernel.h | 322 B | 2024-Nov-02 04:59 |
| PointwiseOpsKernel.cpp | 10.2 KiB | 2024-Oct-31 02:38 |
| PowKernel.cpp | 5.1 KiB | 2024-Oct-31 02:38 |
| README.md | 4.7 KiB | 2024-Oct-31 02:38 |
| RangeFactoriesKernel.cpp | 2.8 KiB | 2024-Oct-31 02:38 |
| Reduce.h | 11.9 KiB | 2024-Oct-31 02:38 |
| ReduceAllOpsKernel.cpp | 8.2 KiB | 2024-Oct-31 02:38 |
| ReduceOpsKernel.cpp | 16.8 KiB | 2024-Nov-02 04:59 |
| ReduceUtils.h | 8.6 KiB | 2024-Oct-31 02:38 |
| RenormKernel.cpp | 1.2 KiB | 2024-Oct-31 02:38 |
| SampledAddmmKernel.cpp | 3.1 KiB | 2024-Oct-31 02:38 |
| SampledAddmmKernel.h | 323 B | 2024-Nov-02 04:59 |
| ScatterGatherKernel.cpp | 37.3 KiB | 2024-Nov-02 04:59 |
| SerialStackImpl.h | 5.3 KiB | 2024-Nov-02 04:59 |
| SoftMaxKernel.cpp | 54.6 KiB | 2024-Nov-02 04:59 |
| SoftmaxKernel.h | 943 B | 2024-Oct-31 02:38 |
| SortingKernel.cpp | 9.1 KiB | 2024-Nov-02 04:59 |
| SparseFactories.cpp | 2.3 KiB | 2024-Nov-02 04:59 |
| SpmmReduceKernel.cpp | 19.7 KiB | 2024-Nov-02 04:59 |
| SpmmReduceKernel.h | 1.3 KiB | 2024-Oct-31 02:38 |
| StackKernel.cpp | 659 B | 2024-Oct-31 02:38 |
| StackKernel.h | 309 B | 2024-Nov-02 04:59 |
| SumKernel.cpp | 22.6 KiB | 2024-Nov-02 04:59 |
| TensorCompareKernel.cpp | 14.1 KiB | 2024-Nov-02 04:59 |
| UnaryOpsKernel.cpp | 35.7 KiB | 2024-Oct-31 02:38 |
| Unfold2d.cpp | 14.9 KiB | 2024-Oct-31 02:38 |
| UnfoldBackwardKernel.cpp | 4.7 KiB | 2024-Oct-31 02:38 |
| UpSampleKernel.cpp | 80.3 KiB | 2024-Nov-02 04:59 |
| UpSampleKernelAVXAntialias.h | 56.8 KiB | 2024-Oct-31 02:38 |
| UpSampleMoreKernel.cpp | 37.7 KiB | 2024-Nov-02 04:59 |
| WeightNormKernel.cpp | 13.8 KiB | 2024-Oct-31 02:38 |
| WeightNormKernel.h | 552 B | 2024-Nov-02 04:59 |
| airy_ai.cpp | 738 B | 2024-Oct-31 02:38 |
| avx_mathfun.h | 17.0 KiB | 2024-Oct-31 02:38 |
| batch_norm_kernel.cpp | 59.7 KiB | 2024-Oct-31 02:38 |
| group_norm_kernel.cpp | 57.9 KiB | 2024-Oct-31 02:38 |
| int4mm_kernel.cpp | 25.3 KiB | 2024-Nov-02 04:59 |
| int8mm_kernel.cpp | 12.3 KiB | 2024-Oct-31 02:38 |
| int_mm_kernel.h | 583 B | 2024-Oct-31 02:38 |
| layer_norm_kernel.cpp | 21.4 KiB | 2024-Oct-31 02:38 |
| mixed_data_type.h | 1.4 KiB | 2024-Nov-02 04:59 |
| moments_utils.h | 6.4 KiB | 2024-Nov-02 04:59 |
| scaled_modified_bessel_k0.cpp | 878 B | 2024-Oct-31 02:38 |
| scaled_modified_bessel_k1.cpp | 878 B | 2024-Oct-31 02:38 |
| spherical_bessel_j0.cpp | 841 B | 2024-Oct-31 02:38 |
| utils.h | 7.0 KiB | 2024-Nov-02 04:59 |
| zmath.h | 6.5 KiB | 2024-Nov-02 04:59 |