Lines Matching refs:ValueRange
62 TileCheck(ValueRange tileSizes) : tileSizes(tileSizes) {} in TileCheck()
75 ValueRange tileSizes;
80 static bool isTiled(AffineExpr expr, ValueRange tileSizes) { in isTiled()
89 static bool isTiled(AffineMap map, ValueRange tileSizes) { in isTiled()
493 [](OpBuilder &b, Location loc, ValueRange args) { in makeMemRefCopyOp()
503 function_ref<scf::ValueVector(OpBuilder &, Location, ValueRange, in doit() argument
504 ValueRange)> in doit()
530 [&](OpBuilder &b, Location loc, ValueRange ivs, ValueRange iterArgs) { in doit()
563 function_ref<scf::ValueVector(OpBuilder &, Location, ValueRange, in doit() argument
564 ValueRange)> in doit()
582 [&](OpBuilder &b, Location loc, ValueRange ivs) { in doit()
611 OpBuilder &b, Location loc, ValueRange lbs, ValueRange ubs, in generateParallelLoopNest()
612 ValueRange steps, ArrayRef<Attribute> iteratorTypes, in generateParallelLoopNest()
613 function_ref<void(OpBuilder &, Location, ValueRange)> bodyBuilderFn, in generateParallelLoopNest()
638 [&](OpBuilder &b, Location loc, ValueRange ivs) { in __anon19d834b50d02()
653 [&](OpBuilder &nestedBuilder, Location nestedLoc, ValueRange localIvs) { in __anon19d834b50e02()
682 [&](OpBuilder &nestedBuilder, Location nestedLoc, ValueRange localIvs) { in __anon19d834b50f02()
707 b.create<scf::YieldOp>(loc, ValueRange{}); in __anon19d834b51002()
728 function_ref<scf::ValueVector(OpBuilder &, Location, ValueRange, in doit() argument
729 ValueRange)> in doit()
777 ValueRange lbs(lbsStorage), ubs(ubsStorage), steps(stepsStorage); in doit()
780 [&](OpBuilder &b, Location loc, ValueRange ivs) { in doit()
791 AffineExpr expr, ValueRange operands) { in fullyComposeAndAffineApply()
800 ValueRange tileSizes, AffineMap map, ValueRange lbs, in makeTiledShape()
801 ValueRange ubs, ValueRange subShapeSizes, in makeTiledShape()
932 ValueRange ivs, ValueRange tileSizes) { in computeTileOffsets()
947 ValueRange tileSizes, in computeTileSizes()
961 SmallVector<Type> getTensorOutputTypes(LinalgOp op, ValueRange operands) { in getTensorOutputTypes()
971 LinalgOp op, ValueRange operands, in insertSlicesBack()
972 ValueRange results) { in insertSlicesBack()
1007 ValueRange ivs, ValueRange tileSizes, in makeTiledShapes()
1070 ValueRange{indexOp.getResult(), offsets[indexOp.dim()]}); in offsetIndices()