Lines Matching defs:matB
677 // b : bufB -> matB
690 Value matB = genAllocCopy(rewriter, loc, bufB, tokens);
708 loc, dnTensorHandleTp, tokenTp, token, matB,
747 token = genDeallocMemRef(rewriter, loc, matB, token);
960 // b : bufB -> matB
965 Value matB = genAllocCopy(rewriter, loc, bufB, tokens);
973 Value szk = linalg::createOrFoldDimOp(rewriter, loc, matB, 0);
986 loc, dnTensorHandleTp, tokenTp, token, matB,
1039 token = genDeallocMemRef(rewriter, loc, matB, token);
1072 // b : bufB -> matB
1081 Value matB = genAllocCopy(rewriter, loc, bufB, tokens);
1102 loc, dnMatHandleTp, tokenTp, token, matB, SmallVector<Value>{szk, szn});
1135 token = genDeallocMemRef(rewriter, loc, matB, token);