Home
last modified time | relevance | path

Searched +path:kernels +path:aten (Results 1 – 25 of 115) sorted by relevance

12345

/aosp_15_r20/external/executorch/kernels/aten/
H A DTARGETS
H A Dfunctions.yaml
H A Dedge_dialect_aten_op.yaml
H A Dtargets.bzl
/aosp_15_r20/external/executorch/kernels/aten/cpu/
H A DTARGETS
/aosp_15_r20/external/executorch/kernels/aten/test/
H A DTARGETS
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp
H A DREADME.md
/aosp_15_r20/external/executorch/kernels/aten/cpu/util/
H A DTARGETS
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/kernels/
H A DcutlassB_f16_aligned_k32_dropout.cu
H A DcutlassB_f16_aligned_k128_dropout.cu
H A DcutlassB_f16_notaligned_k65536.cu
H A DcutlassB_f32_aligned_k65536.cu
H A DcutlassB_bf16_aligned_k65536.cu
/aosp_15_r20/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/kernels/
H A Dflash_fwd_hdim96_fp16_sm80.cu
H A Dflash_bwd_hdim192_bf16_sm80.cu
H A Dflash_fwd_split_hdim256_fp16_sm80.cu
H A Dflash_fwd_split_hdim128_bf16_sm80.cu
H A Dflash_fwd_split_hdim192_fp16_sm80.cu
H A Dflash_bwd_hdim224_bf16_sm80.cu
H A Dflash_fwd_split_hdim160_bf16_sm80.cu
H A Dgenerate_kernels.py
H A Dflash_bwd_hdim64_bf16_sm80.cu
H A Dflash_fwd_hdim128_bf16_sm80.cu
H A Dflash_fwd_split_hdim128_fp16_sm80.cu

12345