| Name | Last modified | Size | Description | |
|---|---|---|---|---|
| Parent Directory | - | |||
| ChannelShuffleKernel.h | 2024-12-02 00:41 | 287 | ||
| CatKernel.h | 2024-12-02 00:41 | 307 | ||
| MaxUnpoolKernel.h | 2024-12-02 00:41 | 308 | ||
| StackKernel.h | 2024-12-02 00:41 | 309 | ||
| CopyKernel.h | 2024-12-02 00:41 | 312 | ||
| PixelShuffleKernel.h | 2024-12-02 00:41 | 322 | ||
| SampledAddmmKernel.h | 2024-12-02 00:41 | 323 | ||
| DepthwiseConvKernel.h | 2024-12-02 00:41 | 471 | ||
| WeightNormKernel.h | 2024-12-02 00:41 | 552 | ||
| int_mm_kernel.h | 2024-12-02 00:41 | 583 | ||
| StackKernel.cpp | 2024-12-02 00:41 | 659 | ||
| airy_ai.cpp | 2024-12-02 00:41 | 738 | ||
| GridSamplerKernel.h | 2024-12-02 00:41 | 825 | ||
| spherical_bessel_j0.cpp | 2024-12-02 00:41 | 841 | ||
| AtomicAddFloat.h | 2024-12-02 00:41 | 857 | ||
| scaled_modified_bessel_k0.cpp | 2024-12-02 00:41 | 878 | ||
| scaled_modified_bessel_k1.cpp | 2024-12-02 00:41 | 878 | ||
| ComplexKernel.cpp | 2024-12-02 00:41 | 925 | ||
| SoftmaxKernel.h | 2024-12-02 00:41 | 943 | ||
| RenormKernel.cpp | 2024-12-02 00:41 | 1.2K | ||
| Intrinsics.h | 2024-12-02 00:41 | 1.2K | ||
| SpmmReduceKernel.h | 2024-12-02 00:41 | 1.3K | ||
| mixed_data_type.h | 2024-12-02 00:41 | 1.4K | ||
| FunctionOfAMatrixUtilsKernel.cpp | 2024-12-02 00:41 | 1.6K | ||
| MaxPooling.cpp | 2024-12-02 00:41 | 1.8K | ||
| CatKernel.cpp | 2024-12-02 00:41 | 2.2K | ||
| SparseFactories.cpp | 2024-12-02 00:41 | 2.3K | ||
| IsContiguous.h | 2024-12-02 00:41 | 2.4K | ||
| LogAddExp.h | 2024-12-02 00:41 | 2.4K | ||
| FillKernel.cpp | 2024-12-02 00:41 | 2.7K | ||
| LinearAlgebraKernel.cpp | 2024-12-02 00:41 | 2.7K | ||
| CrossKernel.cpp | 2024-12-02 00:41 | 2.8K | ||
| RangeFactoriesKernel.cpp | 2024-12-02 00:41 | 2.8K | ||
| IndexKernelUtils.h | 2024-12-02 00:41 | 2.9K | ||
| SampledAddmmKernel.cpp | 2024-12-02 00:41 | 3.1K | ||
| NativeMultiheadAttnKernel.cpp | 2024-12-02 00:41 | 3.6K | ||
| ChannelShuffleKernel.cpp | 2024-12-02 00:41 | 3.7K | ||
| README.md | 2024-12-02 00:41 | 4.7K | ||
| UnfoldBackwardKernel.cpp | 2024-12-02 00:41 | 4.7K | ||
| PowKernel.cpp | 2024-12-02 00:41 | 5.1K | ||
| SerialStackImpl.h | 2024-12-02 00:41 | 5.3K | ||
| moments_utils.h | 2024-12-02 00:41 | 6.4K | ||
| zmath.h | 2024-12-02 00:41 | 6.5K | ||
| LerpKernel.cpp | 2024-12-02 00:41 | 6.5K | ||
| utils.h | 2024-12-02 00:41 | 7.0K | ||
| FusedAdagradKernel.cpp | 2024-12-02 00:41 | 7.0K | ||
| AmpGradScalerKernels.cpp | 2024-12-02 00:41 | 8.1K | ||
| ReduceAllOpsKernel.cpp | 2024-12-02 00:41 | 8.2K | ||
| FusedSGDKernel.cpp | 2024-12-02 00:41 | 8.5K | ||
| ReduceUtils.h | 2024-12-02 00:41 | 8.6K | ||
| MultinomialKernel.cpp | 2024-12-02 00:41 | 8.7K | ||
| MaxUnpoolKernel.cpp | 2024-12-02 00:41 | 8.8K | ||
| PixelShuffleKernel.cpp | 2024-12-02 00:41 | 9.0K | ||
| SortingKernel.cpp | 2024-12-02 00:41 | 9.1K | ||
| PointwiseOpsKernel.cpp | 2024-12-02 00:41 | 10K | ||
| DistributionKernels.cpp | 2024-12-02 00:41 | 11K | ||
| Reduce.h | 2024-12-02 00:41 | 12K | ||
| int8mm_kernel.cpp | 2024-12-02 00:41 | 12K | ||
| HistogramKernel.cpp | 2024-12-02 00:41 | 13K | ||
| CopyKernel.cpp | 2024-12-02 00:41 | 13K | ||
| WeightNormKernel.cpp | 2024-12-02 00:41 | 14K | ||
| TensorCompareKernel.cpp | 2024-12-02 00:41 | 14K | ||
| FusedAdamKernel.cpp | 2024-12-02 00:41 | 14K | ||
| Loops.h | 2024-12-02 00:41 | 15K | ||
| Unfold2d.cpp | 2024-12-02 00:41 | 15K | ||
| BlasKernel.cpp | 2024-12-02 00:41 | 16K | ||
| DistributionTemplates.h | 2024-12-02 00:41 | 16K | ||
| ReduceOpsKernel.cpp | 2024-12-02 00:41 | 17K | ||
| avx_mathfun.h | 2024-12-02 00:41 | 17K | ||
| DistanceOpsKernel.cpp | 2024-12-02 00:41 | 18K | ||
| DepthwiseConvKernel.cpp | 2024-12-02 00:41 | 19K | ||
| SpmmReduceKernel.cpp | 2024-12-02 00:41 | 20K | ||
| layer_norm_kernel.cpp | 2024-12-02 00:41 | 21K | ||
| SumKernel.cpp | 2024-12-02 00:41 | 23K | ||
| int4mm_kernel.cpp | 2024-12-02 00:41 | 25K | ||
| MaxPoolKernel.cpp | 2024-12-02 00:41 | 28K | ||
| PaddingKernel.cpp | 2024-12-02 00:41 | 28K | ||
| IndexKernel.cpp | 2024-12-02 00:41 | 30K | ||
| FlashAttentionKernel.cpp | 2024-12-02 00:41 | 32K | ||
| AdaptiveAvgPoolKernel.cpp | 2024-12-02 00:41 | 32K | ||
| UnaryOpsKernel.cpp | 2024-12-02 00:41 | 36K | ||
| ScatterGatherKernel.cpp | 2024-12-02 00:41 | 37K | ||
| UpSampleMoreKernel.cpp | 2024-12-02 00:41 | 38K | ||
| AdaptiveMaxPoolKernel.cpp | 2024-12-02 00:41 | 39K | ||
| AvgPoolKernel.cpp | 2024-12-02 00:41 | 40K | ||
| BinaryOpsKernel.cpp | 2024-12-02 00:41 | 53K | ||
| SoftMaxKernel.cpp | 2024-12-02 00:41 | 55K | ||
| GridSamplerKernel.cpp | 2024-12-02 00:41 | 56K | ||
| UpSampleKernelAVXAntialias.h | 2024-12-02 00:41 | 57K | ||
| group_norm_kernel.cpp | 2024-12-02 00:41 | 58K | ||
| Activation.cpp | 2024-12-02 00:41 | 59K | ||
| batch_norm_kernel.cpp | 2024-12-02 00:41 | 60K | ||
| UpSampleKernel.cpp | 2024-12-02 00:41 | 80K | ||