Home
last modified time | relevance | path

Searched refs:splitReduction (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp30 FailureOr<SplitReductionResult> mlir::linalg::splitReduction( in splitReduction() function in mlir::linalg
438 return splitReduction(rewriter, op, controlSplitReductionFn, useAlloc); in matchAndRewrite()
/llvm-project/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h1035 splitReduction(RewriterBase &b, LinalgOp op,
1858 /// Patterns to apply `splitReduction` below.
/llvm-project/mlir/include/mlir/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.td1521 `splitReduction` transformation and split factor provided as attribute.
1523 The `splitReduction` transformation splits the first single linalg op
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp2596 : splitReduction(rewriter, target, splitFn, getUseAlloc()); in applyToOne()