Lines Matching refs:CodeGen

56 struct CodeGen {  struct
57 CodeGen(SparsificationOptions o, unsigned numTensors, unsigned numLoops, in CodeGen() argument
423 static Value genVectorReducInit(CodeGen &codegen, OpBuilder &builder, in genVectorReducInit()
448 static Value genVectorReducEnd(CodeGen &codegen, OpBuilder &builder, in genVectorReducEnd()
455 static void updateReduc(Merger &merger, CodeGen &codegen, Value reduc) { in updateReduc()
470 static Value genOutputBuffer(CodeGen &codegen, OpBuilder &builder, in genOutputBuffer()
495 static void genBuffers(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genBuffers()
569 static VectorType vectorType(CodeGen &codegen, Type etp) { in vectorType()
575 static VectorType vectorType(CodeGen &codegen, Value ptr) { in vectorType()
580 static Value genVectorMask(CodeGen &codegen, OpBuilder &builder, Value iv, in genVectorMask()
611 static Value genVectorLoad(CodeGen &codegen, OpBuilder &builder, Value ptr, in genVectorLoad()
628 static void genVectorStore(CodeGen &codegen, OpBuilder &builder, Value rhs, in genVectorStore()
645 static Value genVectorInvariantValue(CodeGen &codegen, OpBuilder &builder, in genVectorInvariantValue()
655 static Value genAffine(CodeGen &codegen, OpBuilder &builder, AffineExpr a, in genAffine()
684 static Value genIndex(CodeGen &codegen, linalg::GenericOp op, OpOperand *t) { in genIndex()
694 static Value genSubscript(CodeGen &codegen, OpBuilder &builder, in genSubscript()
719 static Value genInsertionLoad(CodeGen &codegen, OpBuilder &builder, in genInsertionLoad()
733 static void genInsertionStore(CodeGen &codegen, OpBuilder &builder, in genInsertionStore()
775 static Value genTensorLoad(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genTensorLoad()
797 static void genTensorStore(Merger &merger, CodeGen &codegen, OpBuilder &builder, in genTensorStore()
832 static Value genLoad(CodeGen &codegen, OpBuilder &builder, Location loc, in genLoad()
876 static Value genInvariantValue(Merger &merger, CodeGen &codegen, in genInvariantValue()
885 static Value genAddress(CodeGen &codegen, OpBuilder &builder, Location loc, in genAddress()
897 static Value genIndexValue(CodeGen &codegen, OpBuilder &builder, unsigned idx, in genIndexValue()
933 static Value relinkBranch(CodeGen &codegen, RewriterBase &rewriter, in relinkBranch()
948 static Value genExp(Merger &merger, CodeGen &codegen, RewriterBase &rewriter, in genExp()
972 static bool isInvariantAffine(const CodeGen &codegen, AffineExpr a, in isInvariantAffine()
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()
1124 static bool isVectorFor(CodeGen &codegen, bool isInner, bool isReduction, in isVectorFor()
1144 static bool isParallelFor(CodeGen &codegen, bool isOuter, bool isReduction, in isParallelFor()
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()
1754 CodeGen codegen(options, numTensors, numLoops, sparseOut, outerParNest); in matchAndRewrite()