Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/mem_eff_attention/epilogue/
Depilogue_pipelined.h78 struct ApplyEpilogueOp { struct
79 static CUTLASS_DEVICE typename Op::FragmentOutput apply( in apply()
86 static CUTLASS_DEVICE typename Op::FragmentOutput apply( in apply()