Lines Matching defs:pack
82 // pack transformation.
178 // We can only pack AffineDimExpr atm.
210 /// Return all the pack sizes by which an operand @ `operandPos` is packed.
307 // This pack is just a plain pad.
479 FailureOr<PackResult> linalg::pack(RewriterBase &rewriter,
484 "incorrect number of pack sizes");
517 DBGS() << "++++ After pack size #" << i << ": " << packedSizes[i]
689 return rewriter.notifyMatchFailure(linalgOp, "expect single pack use");
762 /// calling linalg::pack by `mnkPackedSizes`. The order of the packed
781 LLVM_DEBUG(DBGS() << "need 3+ loops to find a matmul to pack, got "
784 linalgOp, "need 3+ loops to find a matmul to pack");
845 LLVM_DEBUG(DBGS() << "Generalized Op to pack: " << genericOp << "\n";);
865 // Add leading zeros to match numLoops, we only pack the last 3 dimensions
893 // calling `pack` would be a good time. One would still need to check that
896 return pack(rewriter, genericOp, adjustedPackedSizes);
1025 /// This method assumes that all outer dims for this pack Op are 1.