Searched refs:distributionMapFn (Results 1 – 2 of 2) sorted by relevance
62 RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn);
324 distributionMapFn(std::move(fn)) {} in WarpOpTransferWrite()339 AffineMap map = distributionMapFn(writeOp); in tryDistributeOp()456 DistributionMapFn distributionMapFn; member938 RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn) { in populateDistributeTransferWriteOpPatterns() argument939 patterns.add<WarpOpTransferWrite>(patterns.getContext(), distributionMapFn); in populateDistributeTransferWriteOpPatterns()