| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgTransforms.cpp | 93 ctx, LinalgTilingOptions().setTileSizes({2000, 3000, 4000}), in applyPatterns()
96 ctx, LinalgTilingOptions().setTileSizes({200, 300, 400}), in applyPatterns()
99 ctx, LinalgTilingOptions().setTileSizes({20, 30, 40}), in applyPatterns()
102 ctx, LinalgTilingOptions().setTileSizes({2, 3, 4}), in applyPatterns()
107 LinalgTilingOptions().setTileSizes({5, 6}).setLoopType( in applyPatterns()
112 ctx, LinalgTilingOptions().setTileSizes(8000), in applyPatterns()
123 LinalgTilingOptions() in applyPatterns()
130 LinalgTilingOptions() in applyPatterns()
136 ctx, LinalgTilingOptions().setTileSizes({20, 30, 40}), in applyPatterns()
141 ctx, LinalgTilingOptions().setTileSizes({5, 6}).setInterchange({1, 0}), in applyPatterns()
[all …]
|
| D | TestLinalgFusionTransforms.cpp | 44 LinalgTilingOptions() in fillFusionPatterns()
57 LinalgTilingOptions() in fillFusionPatterns()
70 LinalgTilingOptions() in fillFusionPatterns()
83 LinalgTilingOptions() in fillFusionPatterns()
98 LinalgTilingOptions().setTileSizes({32, 64}).setLoopType( in fillFusionPatterns()
231 LinalgTilingOptions().setTileSizes(tileSizes).setLoopType( in runOnFunction()
|
| D | TestLinalgCodegenStrategy.cpp | 99 LinalgTilingOptions tilingOptions; in runOnFunction()
103 LinalgTilingOptions registerTilingOptions; in runOnFunction()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 27 struct LinalgTilingOptions;
66 const LinalgTilingOptions &options);
131 const LinalgTilingOptions &tilingOptions);
335 struct LinalgTilingOptions { struct
341 LinalgTilingOptions & argument
349 LinalgTilingOptions &setTileSizes(SmallVector<Value, 4> ts) { in setTileSizes() argument
356 LinalgTilingOptions &setTileSizes(ArrayRef<int64_t> ts);
361 LinalgTilingOptions &setInterchange(ArrayRef<unsigned> interchange) { in setInterchange() argument
369 LinalgTilingOptions &setLoopType(LinalgTilingLoopType lt) { in setLoopType() argument
378 LinalgTilingOptions & argument
[all …]
|
| D | CodegenStrategy.h | 34 explicit Tile(linalg::LinalgTilingOptions options) : options(options) {} in Tile()
45 linalg::LinalgTilingOptions options;
100 CodegenStrategy &tile(linalg::LinalgTilingOptions options) { in tile()
107 CodegenStrategy &tileIf(bool b, linalg::LinalgTilingOptions options) { in tileIf()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Transforms.cpp | 92 LinalgTilingOptions &
93 mlir::linalg::LinalgTilingOptions::setTileSizes(ArrayRef<int64_t> ts) { in setTileSizes()
109 StringRef opName, MLIRContext *context, LinalgTilingOptions options, in LinalgBaseTilingPattern()
115 LinalgTilingOptions options, LinalgMarker marker, PatternBenefit benefit) in LinalgBaseTilingPattern()
155 LinalgTilingOptions tilingOptions, LinalgFusionOptions fusionOptions, in LinalgBaseTileAndFusePattern()
194 LinalgTilingOptions instanceTilingOptions = tilingOptions; in matchAndRewrite()
218 LinalgTilingOptions unfusedTilingOptions = tilingOptions; in matchAndRewrite()
|
| D | Tiling.cpp | 319 const LinalgTilingOptions &options) { in tileLinalgOpImpl()
479 OpBuilder &b, LinalgOp op, const LinalgTilingOptions &options) { in tileLinalgOpImpl()
500 const LinalgTilingOptions &options) { in tileLinalgOp()
539 const LinalgTilingOptions &options, MLIRContext *ctx) {} in insert()
546 const LinalgTilingOptions &options, MLIRContext *ctx) { in insert()
582 const LinalgTilingOptions &options, in insertTilingPatterns()
594 LinalgTilingOptions().setTileSizes(tileSizes).setLoopType(loopType); in applyTilingToLoopPatterns()
|
| D | Fusion.cpp | 796 const LinalgTilingOptions &options, const std::set<unsigned> &fusedLoops) { in tileRootOperation()
802 LinalgTilingOptions tileFusedLoopsOptions = options; in tileRootOperation()
838 const LinalgTilingOptions &tilingOptions) { in tileAndFuseLinalgOpsImpl()
899 const LinalgTilingOptions &tilingOptions) { in tileAndFuseLinalgOps()
|
| D | Vectorization.cpp | 616 context, LinalgTilingOptions().setTileSizes(tileSizes), in populateVectorizationPatterns()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_fuse_linalg.cc | 199 linalg::LinalgTilingOptions() in tileGenericOp()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 319 mlir::linalg::LinalgTilingOptions tilingOptions; in EmitLinalgMatmul()
|