Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/
Dkernel_forward.h498 using DefaultEpilogue = typename DefaultGemm::Epilogue; member
1123 using DefaultEpilogue = typename MM1::DefaultEpilogue; in attention_kernel() local
Dkernel_backward.h419 using DefaultEpilogue = typename DefaultGemm::Epilogue; member
540 using DefaultEpilogue = typename DefaultGemm::Epilogue; member
603 using DefaultEpilogue = typename DefaultGemm::Epilogue; member
2462 using DefaultEpilogue = typename MatmulT::DefaultEpilogue; in accumulateInGmem() local