Searched defs:szk (Results 1 – 1 of 1) sorted by relevance
681 Value szk = linalg::createOrFoldDimOp(rewriter, loc, a, 1); in rewriteSpMM() local 784 Value szk = linalg::createOrFoldDimOp(rewriter, loc, a, 1); in rewriteSpGEMM() local 973 Value szk = linalg::createOrFoldDimOp(rewriter, loc, matB, 0); in rewrite2To4SpMM() local 1076 Value szk in rewriteSDDMM() local [all...]