Searched defs:accumulate_dtype (Results 1 – 2 of 2) sorted by relevance
247 const auto accumulate_dtype = toOpMathType(dtype); in cpu_flash_attention() local502 const auto accumulate_dtype = toOpMathType(dtype); in cpu_flash_attention_backward() local
802 const auto accumulate_dtype = toOpMathType(dtype); in _scaled_dot_product_flash_attention_cpu() local