Searched defs:headSize (Results 1 – 6 of 6) sorted by relevance
207 int64_t headSize = query.size(3); in cpu_flash_attention() local450 int64_t headSize = query.size(3); in cpu_flash_attention_backward() local
678 short headSize = 0; in writeBinaryHeader() local
263 int64_t headSize = query.size(3); in cpu_flash_attention() local
783 int64_t headSize = query.size(3); in _scaled_dot_product_flash_attention_cpu() local
1084 const UInt32 headSize = Get32a(p + 0); in Open2() local
2401 …uint32_t headSize = ( KERNEL_INFO_SIZE_IN_DWORD + numArgs * PER_ARG_SIZE_IN_DWORD ) * sizeof( uint… in CalcKernelDataSize() local