Lines Matching defs:BKIND

324 /// Define ForcedMatmul<ACAT><AKIND><BCAT><BKIND> models.
340 #define MATMUL_INSTANCE(ACAT, AKIND, BCAT, BKIND)                              \
341   struct ForcedMatmul##ACAT##AKIND##BCAT##BKIND                                \
344         ExpandAndQuoteKey(RTNAME(Matmul##ACAT##AKIND##BCAT##BKIND));           \
347 #define MATMUL_DIRECT_INSTANCE(ACAT, AKIND, BCAT, BKIND)
367 #define MATMUL_INSTANCE(ACAT, AKIND, BCAT, BKIND)                              \
368   if (!func && aKind == AKIND && bKind == BKIND &&                             \
373         fir::runtime::getRuntimeFunc<ForcedMatmul##ACAT##AKIND##BCAT##BKIND>(  \
377 #define MATMUL_DIRECT_INSTANCE(ACAT, AKIND, BCAT, BKIND)
394 /// Define ForcedMatmulTranspose<ACAT><AKIND><BCAT><BKIND> models.
395 #define MATMUL_INSTANCE(ACAT, AKIND, BCAT, BKIND)                              \
396   struct ForcedMatmulTranspose##ACAT##AKIND##BCAT##BKIND                       \
399         ExpandAndQuoteKey(RTNAME(MatmulTranspose##ACAT##AKIND##BCAT##BKIND));  \
402 #define MATMUL_DIRECT_INSTANCE(ACAT, AKIND, BCAT, BKIND)
421 #define MATMUL_INSTANCE(ACAT, AKIND, BCAT, BKIND)                              \
423       bCat == TypeCategory::BCAT && bKind == BKIND) {                          \
425         ForcedMatmulTranspose##ACAT##AKIND##BCAT##BKIND>(loc, builder);        \
428 #define MATMUL_DIRECT_INSTANCE(ACAT, AKIND, BCAT, BKIND)