Home
last modified time | relevance | path

Searched defs:accumulate_dtype (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DFlashAttentionKernel.cpp247 const auto accumulate_dtype = toOpMathType(dtype); in cpu_flash_attention() local
502 const auto accumulate_dtype = toOpMathType(dtype); in cpu_flash_attention_backward() local
/external/pytorch/aten/src/ATen/native/transformers/
Dattention.cpp802 const auto accumulate_dtype = toOpMathType(dtype); in _scaled_dot_product_flash_attention_cpu() local