437 checkTilingLegalityImpl(MutableArrayRef<mlir::AffineForOp> origLoops) {  in checkTilingLegalityImpl()  argument
438   assert(!origLoops.empty() && "no original loops provided");  in checkTilingLegalityImpl()
442   origLoops[0]->walk([&](Operation *op) {  in checkTilingLegalityImpl()
448   unsigned numLoops = origLoops.size();  in checkTilingLegalityImpl()
506 checkTilingLegality(MutableArrayRef<mlir::AffineForOp> origLoops) {  in checkTilingLegality()  argument
507   return success(checkTilingLegalityImpl(origLoops));  in checkTilingLegality()
542 void constructTiledLoopNest(MutableArrayRef<AffineForOp> origLoops,  in constructTiledLoopNest()  argument
578   moveLoopBody(origLoops.back(), innermostPointLoop);  in constructTiledLoopNest()
795     MutableArrayRef<AffineForOp> origLoops,  in constructParametricallyTiledIndexSetHyperRect()  argument
797   assert(!origLoops.empty() && "expected atleast one loop in band");  in constructParametricallyTiledIndexSetHyperRect()
798   assert(origLoops.size() == tileSizes.size() &&  in constructParametricallyTiledIndexSetHyperRect()
801   OpBuilder b(origLoops[0].getOperation());  in constructParametricallyTiledIndexSetHyperRect()
802   unsigned width = origLoops.size();  in constructParametricallyTiledIndexSetHyperRect()
806     setInterTileBoundsParametric(b, origLoops[i], newLoops[i], tileSizes[i]);  in constructParametricallyTiledIndexSetHyperRect()
811     setIntraTileBoundsParametric(b, origLoops[i], newLoops[i],  in constructParametricallyTiledIndexSetHyperRect()
822 constructTiledIndexSetHyperRect(MutableArrayRef<AffineForOp> origLoops,  in constructTiledIndexSetHyperRect()  argument
825   assert(!origLoops.empty());  in constructTiledIndexSetHyperRect()
826   assert(origLoops.size() == tileSizes.size());  in constructTiledIndexSetHyperRect()
828   OpBuilder b(origLoops[0].getOperation());  in constructTiledIndexSetHyperRect()
829   unsigned width = origLoops.size();  in constructTiledIndexSetHyperRect()
833     OperandRange newLbOperands = origLoops[i].getLowerBoundOperands();  in constructTiledIndexSetHyperRect()
834     OperandRange newUbOperands = origLoops[i].getUpperBoundOperands();  in constructTiledIndexSetHyperRect()
835     newLoops[i].setLowerBound(newLbOperands, origLoops[i].getLowerBoundMap());  in constructTiledIndexSetHyperRect()
836     newLoops[i].setUpperBound(newUbOperands, origLoops[i].getUpperBoundMap());  in constructTiledIndexSetHyperRect()
841     int64_t largestDiv = getLargestDivisorOfTripCount(origLoops[i]);  in constructTiledIndexSetHyperRect()
842     Optional<uint64_t> mayBeConstantCount = getConstantTripCount(origLoops[i]);  in constructTiledIndexSetHyperRect()
864       AffineBound ub = origLoops[i].getUpperBound();  in constructTiledIndexSetHyperRect()
907   MutableArrayRef<AffineForOp> origLoops = input;  in tilePerfectlyNested()  local
908   AffineForOp rootAffineForOp = origLoops[0];  in tilePerfectlyNested()
915   constructTiledLoopNest(origLoops, rootAffineForOp, width, tiledLoops);  in tilePerfectlyNested()
924   constructTiledIndexSetHyperRect(origLoops, tiledLoops, tileSizes);  in tilePerfectlyNested()
949   MutableArrayRef<AffineForOp> origLoops = input;  in tilePerfectlyNestedParametric()  local
950   AffineForOp rootAffineForOp = origLoops[0];  in tilePerfectlyNestedParametric()
957   constructTiledLoopNest(origLoops, rootAffineForOp, width, tiledLoops);  in tilePerfectlyNestedParametric()
966   constructParametricallyTiledIndexSetHyperRect(origLoops, tiledLoops,  in tilePerfectlyNestedParametric()