Lines Matching refs:AffineMap
43 unsigned pos, AffineMap map, in alignAndAddBound()
49 AffineMap alignedMap = in alignAndAddBound()
57 static AffineMap addConstToResults(AffineMap map, int64_t val) { in addConstToResults()
61 return AffineMap::get(map.getNumDims(), map.getNumSymbols(), newResults, in addConstToResults()
104 canonicalizeMinMaxOp(RewriterBase &rewriter, Operation *op, AffineMap map, in canonicalizeMinMaxOp()
119 AffineMap mapLbUb = isMin ? addConstToResults(map, 1) : map; in canonicalizeMinMaxOp()
126 SmallVector<AffineMap> opLb(1), opUb(1); in canonicalizeMinMaxOp()
128 AffineMap sliceBound = isMin ? opUb[0] : opLb[0]; in canonicalizeMinMaxOp()
134 AffineMap boundMap = isMin ? addConstToResults(sliceBound, -1) : sliceBound; in canonicalizeMinMaxOp()
138 AffineMap alignedBoundMap = boundMap.shiftDims(/*shift=*/1, /*offset=*/dimOp); in canonicalizeMinMaxOp()
183 AffineMap newMap = alignedBoundMap; in canonicalizeMinMaxOp()
251 auto map = AffineMap::get( in addLoopRangeConstraints()
268 Operation *op, AffineMap map, in canonicalizeMinMaxOpInLoop()
322 AffineMap map, ValueRange operands, in rewritePeeledMinMaxOp()