Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DFusionOnTensors.cpp133 OpOperand *iterArg) { in getTiledProducer() argument
175 if (iterArg) { in getTiledProducer()
178 iterArg->set(outputOperand->get()); in getTiledProducer()
229 OpOperand *iterArg = &tileLoop.getOpOperandForRegionIterArg(bbArg); in getTiedBBArgs() local
230 bbArg = iterArg->get().dyn_cast<BlockArgument>(); in getTiedBBArgs()
360 OpOperand *iterArg = nullptr; in fuseProducer() local
363 iterArg = getTiedIterArg(bbArg); in fuseProducer()
365 if (!iterArg || hasOtherUses(bbArg, sliceOp)) in fuseProducer()
367 producerResult = iterArg->get().dyn_cast<OpResult>(); in fuseProducer()
385 tiledProducerLoopIndices, iterArg); in fuseProducer()
/llvm-project-15.0.7/mlir/lib/Target/Cpp/
H A DTranslateToCpp.cpp474 BlockArgument iterArg = std::get<0>(pair); in printOperation() local
476 os << emitter.getOrCreateName(iterArg) << " = " in printOperation()
485 BlockArgument iterArg = std::get<1>(pair); in printOperation() local
488 << emitter.getOrCreateName(iterArg) << ";"; in printOperation()
/llvm-project-15.0.7/flang/lib/Optimizer/Dialect/
H A DFIROps.cpp1839 mlir::BlockArgument fir::IterWhileOp::iterArgToBlockArg(mlir::Value iterArg) { in iterArgToBlockArg() argument
1841 if (iterArg == i.value()) in iterArgToBlockArg()
2128 mlir::BlockArgument fir::DoLoopOp::iterArgToBlockArg(mlir::Value iterArg) { in iterArgToBlockArg() argument
2130 if (iterArg == i.value()) in iterArgToBlockArg()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/SCF/IR/
H A DSCFOps.td288 /// Assert that opOperand is an iterArg.
/llvm-project-15.0.7/flang/include/flang/Optimizer/Dialect/
H A DFIROps.td2109 mlir::BlockArgument iterArgToBlockArg(mlir::Value iterArg);
2251 mlir::BlockArgument iterArgToBlockArg(mlir::Value iterArg);