Lines Matching refs:GenericOp
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()
471 linalg::GenericOp op, MemRefType denseTp, in genOutputBuffer()
496 linalg::GenericOp op) { in genBuffers()
684 static Value genIndex(CodeGen &codegen, linalg::GenericOp op, OpOperand *t) { in genIndex()
695 linalg::GenericOp op, OpOperand *t, in genSubscript()
720 linalg::GenericOp op, OpOperand *t) { in genInsertionLoad()
734 linalg::GenericOp op, OpOperand *t, Value rhs) { in genInsertionStore()
776 linalg::GenericOp op, unsigned exp) { in genTensorLoad()
798 linalg::GenericOp op, unsigned exp, Value rhs) { in genTensorStore()
949 linalg::GenericOp op, unsigned exp, unsigned ldx) { in genExp()
994 linalg::GenericOp op, unsigned exp, unsigned ldx, in genInvariants()
1047 linalg::GenericOp op, unsigned at, bool atStart) { in genExpansion()
1084 linalg::GenericOp op, std::vector<unsigned> &topSort, in genInit()
1169 static bool denseUnitStrides(Merger &merger, linalg::GenericOp op, in denseUnitStrides()
1192 linalg::GenericOp op, bool isOuter, bool isInner, in genFor()
1265 linalg::GenericOp op, unsigned idx, bool needsUniv, in genWhile()
1331 linalg::GenericOp op, std::vector<unsigned> &topSort, in genLoop()
1345 linalg::GenericOp op, std::vector<unsigned> &topSort, in genLocals()
1408 OpBuilder &builder, linalg::GenericOp op, in genWhileInduction()
1474 OpBuilder &builder, linalg::GenericOp op, in genForInduction()
1495 linalg::GenericOp op, unsigned idx, in genIf()
1527 linalg::GenericOp op, scf::IfOp ifOp, Operation *loop, in endIf()
1550 linalg::GenericOp op, std::vector<unsigned> &topSort, in startLoopSeq()
1578 OpBuilder &builder, linalg::GenericOp op, in startLoop()
1593 linalg::GenericOp op, Operation *loop, unsigned idx, in endLoop()
1609 linalg::GenericOp op, unsigned exp, unsigned at, in endLoopSeq()
1628 linalg::GenericOp op, std::vector<unsigned> &topSort, in genStmt()
1687 linalg::GenericOp op) { in genResult()
1710 struct GenericOpSparsifier : public OpRewritePattern<linalg::GenericOp> {
1713 : OpRewritePattern<linalg::GenericOp>(context), options(o) {} in GenericOpSparsifier()
1715 LogicalResult matchAndRewrite(linalg::GenericOp op, in matchAndRewrite()
1764 linalg::GenericOp op) const { in resolveCycle()