Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp202 if (isEmpty() || tiledRootAndFusedOpsLoops.count(rootOp) == 0) in isValid()
206 if (tileLoopOps.size() != tiledRootAndFusedOpsLoops[rootOp].size()) in isValid()
309 if (tiledRootAndFusedOpsLoops.count(rootOp) != 0) { in tileRootOp()
310 tiledRootAndFusedOpsLoops[tiledRootOp->op] = in tileRootOp()
311 tiledRootAndFusedOpsLoops[rootOp]; in tileRootOp()
321 tiledRootAndFusedOpsLoops[rootOp].push_back(tileInterchange[en.index()]); in tileRootOp()
331 if (tiledRootAndFusedOpsLoops.count(consumerOpOperand->getOwner()) == 0) in fuseProducer()
374 consumerOpOperand, tiledRootAndFusedOpsLoops[consumerOp]); in fuseProducer()
386 tiledRootAndFusedOpsLoops[clonedOp] = tiledProducerLoopIndices; in fuseProducer()
410 for (const auto &kvp : tiledRootAndFusedOpsLoops) { in getAllTiledAndFusedOps()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/
H A DUtils.h436 DenseMap<Operation *, SmallVector<int64_t>> tiledRootAndFusedOpsLoops; variable