Home
last modified time | relevance | path

Searched refs:tileLoopOps (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp198 bool TileLoopNest::isEmpty() { return tileLoopOps.empty(); } in isEmpty()
206 if (tileLoopOps.size() != tiledRootAndFusedOpsLoops[rootOp].size()) in isValid()
210 if (rootOp->getParentOp() != tileLoopOps.back()) in isValid()
214 return std::adjacent_find(tileLoopOps.begin(), tileLoopOps.end(), in isValid()
217 }) == tileLoopOps.end(); in isValid()
225 for (auto tileLoop : reverse(tileLoopOps)) { in getTiedBBArgs()
240 if (bbArgs.size() != tileLoopOps.size()) in getTiedIterArg()
242 return &tileLoopOps.front().getOpOperandForRegionIterArg(bbArgs.front()); in getTiedIterArg()
316 tileLoopOps.append(tiledRootOp->loops.begin(), tiledRootOp->loops.end()); in tileRootOp()
405 return tileLoopOps.front()->getOpResults(); in getRootOpReplacementResults()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h406 ArrayRef<scf::ForOp> getLoopOps() { return tileLoopOps; } in getLoopOps()
435 SmallVector<scf::ForOp> tileLoopOps; variable