Searched defs:accum_t (Results 1 – 5 of 5) sorted by relevance
506 Epilogue<scalar_t, accum_t, outscalar_t> epilogue) { in WriteFpropResultsVectorized() argument560 Epilogue<scalar_t, accum_t, outscalar_t> epilogue) { in WriteBpropResultsVectorized() argument620 Epilogue<scalar_t, accum_t, outscalar_t> epilogue) { in WriteFpropResults() argument633 Epilogue<scalar_t, accum_t, outscalar_t> epilogue) { in WriteBpropResults() argument
195 using accum_t = at::opmath_type<scalar_t>; in cpu_flash_attention() typedef435 using accum_t = at::opmath_type<scalar_t>; in cpu_flash_attention_backward() typedef798 using accum_t = at::opmath_type<scalar_t>; in flash_attention_backward_kernel_impl() typedef
255 using accum_t = scalar_t; in cpu_flash_attention() typedef
99 using accum_t = float; member
234 using accum_t = float; member