Lines Matching defs:distributionMapFn

352 /// `distributionMapFn`. Writes of size more than `maxNumElementToExtract`
373       : WarpDistributionPattern(ctx, b), distributionMapFn(std::move(fn)),
389     AffineMap map = distributionMapFn(writeOp.getVector());
568   DistributionMapFn distributionMapFn;
1541       : WarpDistributionPattern(ctx, b), distributionMapFn(std::move(fn)) {}
1565               AffineMap map = distributionMapFn(operand->get());
1658   DistributionMapFn distributionMapFn;
1753     RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn,
1755   patterns.add<WarpOpTransferWrite>(patterns.getContext(), distributionMapFn,
1760     RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn,
1771   patterns.add<WarpOpScfForOp>(patterns.getContext(), distributionMapFn,