Home
last modified time | relevance | path

Searched defs:headSize (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DFlashAttentionKernel.cpp207 int64_t headSize = query.size(3); in cpu_flash_attention() local
450 int64_t headSize = query.size(3); in cpu_flash_attention_backward() local
/external/cldr/tools/cldr-code/src/main/java/org/unicode/cldr/icu/
DLDML2ICUBinaryWriter.java678 short headSize = 0; in writeBinaryHeader() local
/external/executorch/extension/llm/custom_ops/
Dop_sdpa.cpp263 int64_t headSize = query.size(3); in cpu_flash_attention() local
/external/pytorch/aten/src/ATen/native/transformers/
Dattention.cpp783 int64_t headSize = query.size(3); in _scaled_dot_product_flash_attention_cpu() local
/external/lzma/CPP/7zip/Archive/
DDmgHandler.cpp1084 const UInt32 headSize = Get32a(p + 0); in Open2() local
/external/intel-media-driver/media_driver/agnostic/common/cm/
Dcm_kernel_rt.cpp2401 …uint32_t headSize = ( KERNEL_INFO_SIZE_IN_DWORD + numArgs * PER_ARG_SIZE_IN_DWORD ) * sizeof( uint… in CalcKernelDataSize() local