Lines Matching defs:szk

681   Value szk = linalg::createOrFoldDimOp(rewriter, loc, a, 1);
703       genSpMat(rewriter, loc, aTp, spMatHandleTp, tokenTp, token, szm, szk,
709       SmallVector<Value>{szk, szn});
784   Value szk = linalg::createOrFoldDimOp(rewriter, loc, a, 1);
808       genSpMat(rewriter, loc, aTp, spmatHandleTp, tokenTp, token, szm, szk,
813       genSpMat(rewriter, loc, bTp, spmatHandleTp, tokenTp, token, szk, szn,
973   Value szk = linalg::createOrFoldDimOp(rewriter, loc, matB, 0);
981       loc, spMatHandleTp, tokenTp, token, szm, szk,
987       SmallVector<Value>{szk, szn});
1076   Value szk = linalg::createOrFoldDimOp(rewriter, loc, a, 1);
1098       loc, dnMatHandleTp, tokenTp, token, matA, SmallVector<Value>{szm, szk});
1102       loc, dnMatHandleTp, tokenTp, token, matB, SmallVector<Value>{szk, szn});