Searched refs:yieldedValues (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/ |
| H A D | TileUsingInterface.cpp | 292 SmallVector<Value> yieldedValues; in returningMatchAndRewrite() local
309 yieldedValues.push_back(yieldedValue); in returningMatchAndRewrite()
311 return yieldedValues; in returningMatchAndRewrite()
|
| H A D | BufferizableOpInterfaceImpl.cpp | 253 ValueRange yieldedValues, in getEquivalentBuffers() argument
255 unsigned int minSize = std::min(bbArgs.size(), yieldedValues.size()); in getEquivalentBuffers()
259 !yieldedValues[i].getType().isa<TensorType>()) in getEquivalentBuffers()
261 if (state.areEquivalentBufferizedValues(bbArgs[i], yieldedValues[i])) in getEquivalentBuffers()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/SCF/Utils/ |
| H A D | Utils.cpp | 369 ValueRange iterArgs, ValueRange yieldedValues) { in generateUnrolledLoop() argument
382 SmallVector<Value, 4> lastYielded(yieldedValues); in generateUnrolledLoop()
405 lastYielded[i] = operandMap.lookup(yieldedValues[i]); in generateUnrolledLoop()
522 auto yieldedValues = forOp.getBody()->getTerminator()->getOperands(); in loopUnrollByFactor() local
532 annotateFn, iterArgs, yieldedValues); in loopUnrollByFactor()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 1006 ValueRange iterArgs, ValueRange yieldedValues) { in generateUnrolledLoop() argument
1019 SmallVector<Value, 4> lastYielded(yieldedValues); in generateUnrolledLoop()
1042 lastYielded[i] = operandMap.lookup(yieldedValues[i]); in generateUnrolledLoop()
1128 auto yieldedValues = forOp.getBody()->getTerminator()->getOperands(); in loopUnrollByFactor() local
1142 /*iterArgs=*/iterArgs, /*yieldedValues=*/yieldedValues); in loopUnrollByFactor()
|