Searched defs:CONST_NAME (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/transformers/cuda/flash_attn/ |
| D | static_switch.h | 18 #define BOOL_SWITCH(COND, CONST_NAME, ...) \ argument
30 #define DROPOUT_SWITCH(COND, CONST_NAME, ...) \ argument
40 #define ALIBI_SWITCH(COND, CONST_NAME, ...) \ argument
50 #define EVENK_SWITCH(COND, CONST_NAME, ...) \ argument
60 #define LOCAL_SWITCH(COND, CONST_NAME, ...) \ argument
|
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TriangularOps.cu | 22 #define BOOL_SWITCH(COND, CONST_NAME, ...) \ argument
|