Lines Matching refs:Merger
160 static bool findAffine(Merger &merger, unsigned tensor, AffineExpr a, Dim dim, in findAffine()
190 static bool findSparseAnnotations(Merger &merger, linalg::GenericOp op) { in findSparseAnnotations()
257 static bool computeIterationGraph(Merger &merger, linalg::GenericOp op, in computeIterationGraph()
322 static bool isAdmissableTensorExp(Merger &merger, linalg::GenericOp op, in isAdmissableTensorExp()
455 static void updateReduc(Merger &merger, CodeGen &codegen, Value reduc) { in updateReduc()
495 static void genBuffers(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genBuffers()
775 static Value genTensorLoad(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genTensorLoad()
797 static void genTensorStore(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genTensorStore()
876 static Value genInvariantValue(Merger &merger, CodeGen &codegen, in genInvariantValue()
948 static Value genExp(Merger &merger, CodeGen &codegen, RewriterBase &rewriter, in genExp()
993 static void genInvariants(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genInvariants()
1046 static void genExpansion(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genExpansion()
1083 static bool genInit(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genInit()
1169 static bool denseUnitStrides(Merger &merger, linalg::GenericOp op, in denseUnitStrides()
1191 static Operation *genFor(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genFor()
1264 static Operation *genWhile(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genWhile()
1330 static Operation *genLoop(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genLoop()
1344 static void genLocals(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genLocals()
1407 static void genWhileInduction(Merger &merger, CodeGen &codegen, in genWhileInduction()
1473 static void genForInduction(Merger &merger, CodeGen &codegen, in genForInduction()
1494 static scf::IfOp genIf(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genIf()
1526 static void endIf(Merger &merger, CodeGen &codegen, OpBuilder &builder, in endIf()
1549 static bool startLoopSeq(Merger &merger, CodeGen &codegen, OpBuilder &builder, in startLoopSeq()
1577 static Operation *startLoop(Merger &merger, CodeGen &codegen, in startLoop()
1592 static bool endLoop(Merger &merger, CodeGen &codegen, OpBuilder &builder, in endLoop()
1608 static void endLoopSeq(Merger &merger, CodeGen &codegen, OpBuilder &builder, in endLoopSeq()
1627 static void genStmt(Merger &merger, CodeGen &codegen, RewriterBase &rewriter, in genStmt()
1686 static void genResult(Merger &merger, CodeGen &codegen, RewriterBase &rewriter, in genResult()
1722 Merger merger(numTensors, numLoops); in matchAndRewrite()
1763 LogicalResult resolveCycle(Merger &merger, PatternRewriter &rewriter, in resolveCycle()