Home
last modified time | relevance | path

Searched refs:allIvs (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DLoops.cpp84 makeCanonicalAffineApplies(b, loc, maps[0], allIvs), in getInputAndOutputIndices()
85 makeCanonicalAffineApplies(b, loc, maps[2], allIvs)}; in getInputAndOutputIndices()
127 auto allIvsPlusDims = SmallVector<Value>(allIvs.begin(), allIvs.end()); in emitScalarImplementation()
170 SmallVector<Value> allIvs; in replaceIndexOpsByInductionVariables() local
178 allIvs.push_back(forOp.getInductionVar()); in replaceIndexOpsByInductionVariables()
181 allIvs.push_back(affineForOp.getInductionVar()); in replaceIndexOpsByInductionVariables()
185 assert(linalgOp.getNumLoops() == allIvs.size() && in replaceIndexOpsByInductionVariables()
192 rewriter.replaceOp(indexOp, allIvs[indexOp.dim()]); in replaceIndexOpsByInductionVariables()
214 SmallVector<Value> allIvs; in linalgOpToLoopsImpl() local
221 allIvs.append(ivs.begin(), ivs.end()); in linalgOpToLoopsImpl()
[all …]
H A DFusion.cpp170 SmallVector<Value> allIvs; in fuse() local
171 llvm::transform(loopRanges, std::back_inserter(allIvs), in fuse()
173 offsetIndices(b, clonedOp, allIvs); in fuse()
H A DFusionOnTensors.cpp156 SmallVector<Value> allIvs(producerOp.getNumLoops(), nullptr); in getTiledProducer() local
162 allIvs[tiledProducerLoop] = tileIvs[tiledProducerLoop]; in getTiledProducer()
189 offsetIndices(b, clonedOp, allIvs); in getTiledProducer()
H A DTiling.cpp77 SmallVector<Value> allIvs(op.getNumLoops(), nullptr); in transformIndexOps() local
78 for (auto &en : enumerate(allIvs)) { in transformIndexOps()
84 offsetIndices(b, op, allIvs); in transformIndexOps()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/
H A DAffineCanonicalizationUtils.cpp272 DenseSet<Value> allIvs; in canonicalizeMinMaxOpInLoop() local
277 if (llvm::is_contained(allIvs, operand)) in canonicalizeMinMaxOpInLoop()
286 allIvs.insert(iv); in canonicalizeMinMaxOpInLoop()