Lines Matching defs:BlockPackMatmul
1 //===- BlockPackMatmul.cpp - Linalg matmul block packing ------------------===//
218 struct BlockPackMatmul : public OpRewritePattern<OpTy> {
219 BlockPackMatmul(MLIRContext *context, ControlBlockPackMatmulFn fun,
237 struct BlockPackMatmul<linalg::GenericOp>
239 BlockPackMatmul(MLIRContext *context, ControlBlockPackMatmulFn fun,
312 patterns.add<BlockPackMatmul<linalg::GenericOp>,
313 BlockPackMatmul<linalg::MatmulOp>,
314 BlockPackMatmul<linalg::BatchMatmulOp>,
315 BlockPackMatmul<linalg::MatmulTransposeAOp>,
316 BlockPackMatmul<linalg::BatchMatmulTransposeAOp>,
317 BlockPackMatmul<linalg::MatmulTransposeBOp>,
318 BlockPackMatmul<linalg::BatchMatmulTransposeBOp>>(