Searched refs:DistributionMapFn (Results 1 – 2 of 2) sorted by relevance
42 using DistributionMapFn = std::function<AffineMap(vector::TransferWriteOp)>; variable62 RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn);
321 WarpOpTransferWrite(MLIRContext *ctx, DistributionMapFn fn, in WarpOpTransferWrite()456 DistributionMapFn distributionMapFn;938 RewritePatternSet &patterns, const DistributionMapFn &distributionMapFn) { in populateDistributeTransferWriteOpPatterns()