Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/
Dstatic_switch.h30 #define DROPOUT_SWITCH(COND, CONST_NAME, ...) \ macro
36 #define DROPOUT_SWITCH BOOL_SWITCH macro