Lines Matching defs:packedMatmul

169   FailureOr<PackResult> packedMatmul = packMatmulGreedily(
172   if (failed(packedMatmul))
175   assert(packedMatmul->packOps.size() == 3 &&
177   assert(packedMatmul->unPackOps.size() == 1 &&
181       inferContractionDims(packedMatmul->packedLinalgOp);
186       dyn_cast<linalg::GenericOp>(packedMatmul->packedLinalgOp.getOperation());
191       rewriter, packedMatmul->packedLinalgOp, packedMatmul->packOps[0], maps[0],
198   packedMatmul->packOps[0] = packedLhs->transposedPackOp;
199   packedMatmul->packedLinalgOp = packedLhs->transposedLinalgOp;
203       rewriter, packedMatmul->packedLinalgOp, packedMatmul->packOps[1], maps[1],
210   packedMatmul->packOps[1] = packedRhs->transposedPackOp;
211   packedMatmul->packedLinalgOp = packedRhs->transposedLinalgOp;
213   return packedMatmul;
225     FailureOr<PackResult> packedMatmul =
227     if (failed(packedMatmul))
267     FailureOr<PackResult> packedMatmul =
269     if (failed(packedMatmul))