Searched refs:LinalgTilingLoopType (Results 1 – 6 of 6) sorted by relevance
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgFusionTransforms.cpp | 46 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
59 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
72 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
85 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
99 LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
232 LinalgTilingLoopType::ParallelLoops)); in runOnFunction()
|
| D | TestLinalgTransforms.cpp | 108 LinalgTilingLoopType::ParallelLoops), in applyPatterns()
150 .setLoopType(LinalgTilingLoopType::ParallelLoops), in applyPatterns()
329 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
345 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
361 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
377 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
393 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
410 .setLoopType(LinalgTilingLoopType::ParallelLoops) in fillTileAndDistributePatterns()
426 .setLoopType(LinalgTilingLoopType::Loops) in fillTileAndDistributePatterns()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Tiling.cpp | 502 case LinalgTilingLoopType::Loops: in tileLinalgOp()
504 case LinalgTilingLoopType::ParallelLoops: in tileLinalgOp()
590 static void applyTilingToLoopPatterns(LinalgTilingLoopType loopType, in applyTilingToLoopPatterns()
613 applyTilingToLoopPatterns(LinalgTilingLoopType::Loops, getFunction(), in runOnFunction()
626 applyTilingToLoopPatterns(LinalgTilingLoopType::ParallelLoops, in runOnFunction()
|
| D | Fusion.cpp | 901 case LinalgTilingLoopType::Loops: in tileAndFuseLinalgOps()
904 case LinalgTilingLoopType::ParallelLoops: in tileAndFuseLinalgOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_fuse_linalg.cc | 196 ? linalg::LinalgTilingLoopType::ParallelLoops in tileGenericOp()
197 : linalg::LinalgTilingLoopType::Loops; in tileGenericOp()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 326 enum class LinalgTilingLoopType { enum
367 LinalgTilingLoopType loopType = LinalgTilingLoopType::Loops;
369 LinalgTilingOptions &setLoopType(LinalgTilingLoopType lt) { in setLoopType()
|