Searched refs:tensorResults (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Tiling.cpp | 393 tiledOp.tensorResults.assign(tiledOp.op->result_begin(), in tileLinalgOpImpl()
443 SmallVector<Value, 4> ivs, tensorResults; in tileLinalgOpImpl() local
473 tensorResults = in tileLinalgOpImpl()
475 return scf::ValueVector(tensorResults.begin(), tensorResults.end()); in tileLinalgOpImpl()
506 res, loops, outermostLoop ? outermostLoop->getResults() : tensorResults}; in tileLinalgOpImpl()
|
| H A D | Transforms.cpp | 346 if (res.tensorResults.size() == loopOp->getNumResults() && in peelTiledLinalgOp()
347 std::equal(res.tensorResults.begin(), res.tensorResults.end(), in peelTiledLinalgOp()
349 res.tensorResults = loopResults; in peelTiledLinalgOp()
383 if (res->tensorResults.empty()) in returningMatchAndRewrite()
386 rewriter.replaceOp(op, res->tensorResults); in returningMatchAndRewrite()
|
| H A D | FusionOnTensors.cpp | 306 rootOp->replaceAllUsesWith(tiledRootOp->tensorResults); in tileRootOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 973 SmallVector<Value> tensorResults; in insertSlicesBack() local
974 tensorResults.reserve(results.size()); in insertSlicesBack()
987 tensorResults.push_back(inserted); in insertSlicesBack()
989 tensorResults.push_back(results[resultIdx]); in insertSlicesBack()
993 return tensorResults; in insertSlicesBack()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 159 SmallVector<Value, 4> tensorResults; member
|