Lines Matching refs:ubMap
251 AffineMap lbMap, AffineMap ubMap, in generateShiftedLoop() argument
258 assert(ubMap.getNumInputs() == ubOperands.size()); in generateShiftedLoop()
261 ubOperands, ubMap, srcForOp.getStep()); in generateShiftedLoop()
682 AffineMap ubMap = in setIntraTileBoundsParametric() local
685 newIntraTileLoop.setUpperBound(ubOperands, ubMap); in setIntraTileBoundsParametric()
779 AffineMap ubMap = in setInterTileBoundsParametric() local
782 newLoop.setUpperBound(ubOperands, ubMap); in setInterTileBoundsParametric()
852 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
855 /*operands=*/newLoops[i].getInductionVar(), ubMap); in constructTiledIndexSetHyperRect()
885 AffineMap ubMap = in constructTiledIndexSetHyperRect() local
888 newLoops[width + i].setUpperBound(/*operands=*/ubOperands, ubMap); in constructTiledIndexSetHyperRect()
892 AffineMap ubMap = AffineMap::get(1, 0, dim + tileSizes[i]); in constructTiledIndexSetHyperRect() local
893 newLoops[width + i].setUpperBound(newLoops[i].getInductionVar(), ubMap); in constructTiledIndexSetHyperRect()
1663 auto ubMap = forOp.getUpperBoundMap(); in stripmineSink() local
1665 augmentMapAndBounds(b, forOp.getInductionVar(), &ubMap, &ubOperands, in stripmineSink()
1674 ubOperands, ubMap, originalStep); in stripmineSink()
2227 assert(llvm::all_of(ubMaps, [&](AffineMap ubMap) { in generatePointWiseCopy() argument
2228 return ubMap.getNumInputs() == ubOperands.size(); in generatePointWiseCopy()
2909 ValueRange ubOperands, AffineMap ubMap, int64_t step) { in createCanonicalizedAffineForOp() argument
2916 fullyComposeAffineMapAndOperands(&ubMap, &upperOperands); in createCanonicalizedAffineForOp()
2917 canonicalizeMapAndOperands(&ubMap, &upperOperands); in createCanonicalizedAffineForOp()
2918 ubMap = removeDuplicateExprs(ubMap); in createCanonicalizedAffineForOp()
2920 return b.create<AffineForOp>(loc, lowerOperands, lbMap, upperOperands, ubMap, in createCanonicalizedAffineForOp()