Searched refs:tilingOptions (Results 1 – 5 of 5) sorted by relevance
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgCodegenStrategy.cpp | 99 LinalgTilingOptions tilingOptions; in runOnFunction() local
101 tilingOptions = tilingOptions.setTileSizes(tileSizes); in runOnFunction()
124 strategy.tileIf<MatmulOp>(!tileSizes.empty(), tilingOptions) in runOnFunction()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Transforms.cpp | 155 LinalgTilingOptions tilingOptions, LinalgFusionOptions fusionOptions, in LinalgBaseTileAndFusePattern() argument
159 dependenceGraph(dependenceGraph), tilingOptions(tilingOptions),
193 tilingOptions.tileSizeComputationFunction(rewriter, op); in matchAndRewrite()
194 LinalgTilingOptions instanceTilingOptions = tilingOptions; in matchAndRewrite()
218 LinalgTilingOptions unfusedTilingOptions = tilingOptions; in matchAndRewrite()
|
| D | Fusion.cpp | 838 const LinalgTilingOptions &tilingOptions) { in tileAndFuseLinalgOpsImpl() argument
853 if (!tilingOptions.interchangeVector.empty()) { in tileAndFuseLinalgOpsImpl()
880 tilingOptions.tileSizeComputationFunction(builder, rootOp); in tileAndFuseLinalgOpsImpl()
882 builder, rootOp, tileSizeVector, tilingOptions, ret.fusedLoopDims); in tileAndFuseLinalgOpsImpl()
899 const LinalgTilingOptions &tilingOptions) { in tileAndFuseLinalgOps() argument
900 switch (tilingOptions.loopType) { in tileAndFuseLinalgOps()
903 tilingOptions); in tileAndFuseLinalgOps()
906 builder, ops, dependenceGraph, tilingOptions); in tileAndFuseLinalgOps()
|
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 131 const LinalgTilingOptions &tilingOptions);
447 LinalgTilingOptions tilingOptions,
460 LinalgTilingOptions tilingOptions; member
479 LinalgTilingOptions tilingOptions,
486 OpTy::getOperationName(), context, dependenceGraph, tilingOptions, in LinalgBaseTileAndFusePattern()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 319 mlir::linalg::LinalgTilingOptions tilingOptions; in EmitLinalgMatmul() local
320 tilingOptions = tilingOptions.setTileSizes(GetMlirGemmTileSize()); in EmitLinalgMatmul()
325 strategy.tile<mlir::linalg::GenericOp>(tilingOptions) in EmitLinalgMatmul()
|