Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DConstantFold.cpp285 auto yieldArg = yieldVal.dyn_cast<BlockArgument>(); in getRegionComputeFn() local
286 if (!yieldArg || yieldArg.getOwner() != &body) in getRegionComputeFn()
288 if (yieldArg.getArgNumber() != 0) in getRegionComputeFn()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgOps.cpp1138 auto yieldArg = yieldVal.value().dyn_cast<BlockArgument>(); in matchAndRewrite() local
1139 if (!yieldArg || yieldArg.getOwner() != &body) in matchAndRewrite()
1141 unsigned argumentNumber = yieldArg.getArgNumber(); in matchAndRewrite()