Lines Matching defs:ValueRange

248 getTileOffsetAndSizes(RewriterBase &rewriter, Location loc, ValueRange ivs,
369 RewriterBase &rewriter, Location loc, ValueRange ivs, ValueRange newBbArgs,
378 ValueRange newDestArgs) {
398 ArrayRef<OpFoldResult> tileSizes, ValueRange destinationTensors,
421 Value iv, ValueRange /*iterArgs*/) {});
480 ArrayRef<Attribute> mappingVector, ValueRange destinationTensors,
552 ArrayRef<OpFoldResult> numThreads, ValueRange destinationTensors,
559 return tiledBodyFn(rewriter, loc, ValueRange{}, destinationTensors,
613 ValueRange regionIterArg, ArrayRef<OpFoldResult> offsets,
687 ValueRange partialResults,
730 ValueRange newInitOperands,
738 scf::ForOp loopOp, RewriterBase &rewriter, ValueRange newInitOperands,
748 inits, [](OpBuilder &, Location, Value, ValueRange) {});
762 ValueRange newRegionIterArgs =
792 scf::ForallOp loopOp, RewriterBase &rewriter, ValueRange newInitOperands,
802 [](OpBuilder &, Location, ValueRange) {});
815 ValueRange regionIterArgs =
847 ValueRange newInitOperands, YieldTiledValuesFn yieldTiledValuesFn) {
867 ValueRange newInitValues, YieldTiledValuesFn getNewTiledYieldsFn) {
887 [&](OpBuilder &b, Location loc, Value iv, ValueRange iterArgs) {});
924 ValueRange additionalYields =
978 [&](RewriterBase &rewriter, Location loc, ValueRange ivs,
979 ValueRange regionIterArgs, SmallVector<Value> &tiledResults,
1283 [&](RewriterBase &innerRewriter, Location loc, ValueRange /*ivs*/,
1284 ValueRange newRegionIterArgs, SmallVector<Value> &tiledResult,
1412 void notifyOperationReplaced(Operation *op, ValueRange replacement) override;
1476 ValueRange replacement) {
1493 void updateReplacementValues(ValueRange origValues,
1494 ValueRange replaceValues) {
1511 void notifyOperationReplaced(Operation *op, ValueRange values) override {
2080 [&](RewriterBase &innerRewriter, Location loc, ValueRange /*ivs*/,
2081 ValueRange newRegionIterArgs, SmallVector<Value> &tiledResult,
2215 strideVal, ValueRange{});