Searched refs:tileInterchange (Results 1 – 5 of 5) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | FusionOnTensors.cpp | 277 ArrayRef<int64_t> tileInterchange, in tileRootOp() argument
287 tileInterchange.begin(), tileInterchange.end())) in tileRootOp()
321 tiledRootAndFusedOpsLoops[rootOp].push_back(tileInterchange[en.index()]); in tileRootOp()
425 ArrayRef<int64_t> tileInterchange, in tileConsumerAndFuseProducers() argument
427 assert(tileSizes.size() == tileInterchange.size() && in tileConsumerAndFuseProducers()
429 assert(isPermutation(tileInterchange) && in tileConsumerAndFuseProducers()
439 applyPermutationToVector(iterTypes, tileInterchange); in tileConsumerAndFuseProducers()
463 for (int64_t i : tileInterchange.take_front(split)) in tileConsumerAndFuseProducers()
465 for (int64_t i : tileInterchange.drop_front(split)) in tileConsumerAndFuseProducers()
469 if (failed(tileLoopNest.tileRootOp(b, outerTileSizes, tileInterchange, in tileConsumerAndFuseProducers()
[all …]
|
| H A D | Transforms.cpp | 489 if (!options.tileInterchange.empty() && in returningMatchAndRewrite()
490 options.tileInterchange.size() != options.tileSizes.size()) in returningMatchAndRewrite()
499 options.tileInterchange.empty() in returningMatchAndRewrite()
501 : SmallVector<int64_t>(options.tileInterchange.begin(), in returningMatchAndRewrite()
502 options.tileInterchange.begin() + in returningMatchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 389 ArrayRef<int64_t> tileInterchange,
444 ArrayRef<int64_t> tileInterchange,
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | Transforms.h | 579 SmallVector<int64_t> tileInterchange; member
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 169 fusionOptions.tileInterchange = extractFromI64ArrayAttr(getTileInterchange()); in apply()
|