Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp61 FailureOr<LinalgOp> mlir::linalg::splitReduction( in splitReduction() function in mlir::linalg
71 splitReduction(b, op, controlSplitReductionFn, useAlloc); in splitReduction()
81 FailureOr<SplitReductionResult> mlir::linalg::splitReduction( in splitReduction() function in mlir::linalg
463 return splitReduction(rewriter, op, controlSplitReductionFn, filter, in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/
H A DTransforms.h1444 splitReduction(PatternRewriter &b, LinalgOp op,
1458 splitReduction(PatternRewriter &b, LinalgOp op,
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.td453 `splitReduction` transformation and split factor provided as attribute.
455 The `splitReduction` transformation splits the first single linalg op
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp840 : splitReduction(rewriter, target, splitFn, getUseAlloc()); in applyToOne()