Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp89 SmallVector<int64_t> tiledProducerLoopIndices; in getTiledProducerLoops() local
92 std::back_inserter(tiledProducerLoopIndices), [&](unsigned idx) { in getTiledProducerLoops()
96 return tiledProducerLoopIndices; in getTiledProducerLoops()
132 ArrayRef<int64_t> tiledProducerLoopIndices, in getTiledProducer() argument
157 for (auto it : zip(tiledSliceDimIndices, tiledProducerLoopIndices)) { in getTiledProducer()
379 SmallVector<int64_t> tiledProducerLoopIndices = in fuseProducer() local
385 tiledProducerLoopIndices, iterArg); in fuseProducer()
386 tiledRootAndFusedOpsLoops[clonedOp] = tiledProducerLoopIndices; in fuseProducer()