Home
last modified time | relevance | path

Searched refs:setLoopType (Results 1 – 5 of 5) sorted by relevance

/external/llvm-project/mlir/test/lib/Transforms/
DTestLinalgFusionTransforms.cpp46 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
59 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
72 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
85 .setLoopType(LinalgTilingLoopType::ParallelLoops), in fillFusionPatterns()
98 LinalgTilingOptions().setTileSizes({32, 64}).setLoopType( in fillFusionPatterns()
231 LinalgTilingOptions().setTileSizes(tileSizes).setLoopType( in runOnFunction()
DTestLinalgTransforms.cpp107 LinalgTilingOptions().setTileSizes({5, 6}).setLoopType( 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/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dlhlo_fuse_linalg.cc201 .setLoopType(loopType)); in tileGenericOp()
/external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
DTransforms.h369 LinalgTilingOptions &setLoopType(LinalgTilingLoopType lt) { in setLoopType() function
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DTiling.cpp594 LinalgTilingOptions().setTileSizes(tileSizes).setLoopType(loopType); in applyTilingToLoopPatterns()