Lines Matching refs:forOps
821 SmallVectorImpl<T> &forOps, T rootForOp, in getPerfectlyNestedLoopsImpl() argument
824 forOps.push_back(rootForOp); in getPerfectlyNestedLoopsImpl()
885 SmallVector<Loops, 8> mlir::tile(ArrayRef<scf::ForOp> forOps, in tile() argument
890 for (auto it : llvm::zip(forOps, sizes)) { in tile()
898 Loops mlir::tile(ArrayRef<scf::ForOp> forOps, ArrayRef<Value> sizes, in tile() argument
901 for (auto loops : tile(forOps, sizes, ArrayRef<scf::ForOp>(target))) { in tile()
911 SmallVector<scf::ForOp, 4> forOps; in tilePerfectlyNested() local
912 forOps.reserve(sizes.size()); in tilePerfectlyNested()
913 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in tilePerfectlyNested()
914 if (forOps.size() < sizes.size()) in tilePerfectlyNested()
915 sizes = sizes.take_front(forOps.size()); in tilePerfectlyNested()
917 return ::tile(forOps, sizes, forOps.back()); in tilePerfectlyNested()
929 SmallVector<scf::ForOp, 4> forOps; in extractFixedOuterLoops() local
930 forOps.reserve(sizes.size()); in extractFixedOuterLoops()
931 getPerfectlyNestedLoopsImpl(forOps, rootForOp, sizes.size()); in extractFixedOuterLoops()
932 if (forOps.size() < sizes.size()) in extractFixedOuterLoops()
933 sizes = sizes.take_front(forOps.size()); in extractFixedOuterLoops()
944 auto forOp = forOps[i]; in extractFixedOuterLoops()
956 auto intraTile = tile(forOps, tileSizes, forOps.back()); in extractFixedOuterLoops()
957 TileLoops tileLoops = std::make_pair(forOps, intraTile); in extractFixedOuterLoops()