Searched refs:LinalgPromotionOptions (Results 1 – 8 of 8) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| D | Transforms.h | 165 struct LinalgPromotionOptions { struct
168 LinalgPromotionOptions &setOperandsToPromote(ArrayRef<int64_t> operands) { in setOperandsToPromote() argument
180 LinalgPromotionOptions &setUseFullTileBuffers(ArrayRef<bool> useFullTiles) { in setUseFullTileBuffers() argument
191 LinalgPromotionOptions &setUseFullTileBuffersByDefault(bool use) { in setUseFullTileBuffersByDefault() argument
197 LinalgPromotionOptions &setDynamicBuffers(unsigned dynamic) { in setDynamicBuffers() argument
203 LinalgPromotionOptions &setAlignment(unsigned align) { in setAlignment() argument
209 LinalgPromotionOptions &setUseAlloca(bool use) { in setUseAlloca() argument
218 LinalgPromotionOptions & argument
229 LinalgPromotionOptions &setCopyInOutFns(CopyCallbackFn const ©In, in setCopyInOutFns() argument
262 LinalgPromotionOptions options,
[all …]
|
| D | CodegenStrategy.h | 52 explicit Promote(linalg::LinalgPromotionOptions options) : options(options) {} in Promote()
63 linalg::LinalgPromotionOptions options;
113 CodegenStrategy &promote(linalg::LinalgPromotionOptions options) { in promote()
120 CodegenStrategy &promoteIf(bool b, linalg::LinalgPromotionOptions options) { in promoteIf()
|
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Promotion.cpp | 70 static Value allocBuffer(const LinalgPromotionOptions &options, in allocBuffer()
107 const LinalgPromotionOptions &options, OpBuilder &builder, in defaultAllocBufferCallBack()
134 defaultDeallocBufferCallBack(const LinalgPromotionOptions &options, in defaultDeallocBufferCallBack()
151 const LinalgPromotionOptions &options);
172 LinalgOp linalgOp, const LinalgPromotionOptions &options) in LinalgOpInstancePromotionOptions()
368 LinalgPromotionOptions options) { in promoteSubviewsPrecondition()
389 LinalgPromotionOptions options, in promoteSubViews()
407 auto options = LinalgPromotionOptions() in runOnFunction()
|
| D | Transforms.cpp | 270 StringRef opName, MLIRContext *context, LinalgPromotionOptions options, in LinalgBasePromotionPattern()
|
| D | Vectorization.cpp | 620 context, LinalgPromotionOptions().setUseFullTileBuffersByDefault(true), in populateVectorizationPatterns()
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgCodegenStrategy.cpp | 126 LinalgPromotionOptions() in runOnFunction()
130 .promoteIf<MatmulOp>(registerPromote, LinalgPromotionOptions() in runOnFunction()
|
| D | TestLinalgTransforms.cpp | 192 ctx, LinalgPromotionOptions().setUseFullTileBuffersByDefault(true), in applyPatterns()
197 LinalgPromotionOptions() in applyPatterns()
204 LinalgPromotionOptions() in applyPatterns()
231 ctx, LinalgPromotionOptions().setUseFullTileBuffersByDefault(true), in fillL1TilingAndMatmulToVectorPatterns()
289 LinalgPromotionOptions() in fillPromotionCallBackPatterns()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 327 mlir::linalg::LinalgPromotionOptions() in EmitLinalgMatmul()
|