Searched refs:insertSplitDimension (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 39 unsigned insertSplitDimension = control.index; in splitReduction() local
50 insertSplitDimension = reductionDim + 1; in splitReduction()
90 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
91 exprs.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
95 exprs.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
97 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
104 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
132 outputExpr.push_back(b.getAffineDimExpr(insertSplitDimension)); in splitReduction()
138 b.getAffineDimExpr(dim < insertSplitDimension ? dim : dim + 1)); in splitReduction()
162 if (insertSplitDimension == index) in splitReduction()
[all …]
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 2557 int64_t splitFactor, int64_t insertSplitDimension, bool innerParallel, in applyToOne()
2565 builder.getI64IntegerAttr(insertSplitDimension)); in applyToOne()
2525 build(OpBuilder & builder,OperationState & result,Value target,int64_t splitFactor,int64_t insertSplitDimension,bool innerParallel,bool useScalingAlgorithm,bool useAlloc) build() argument
|
| /llvm-project/mlir/include/mlir/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.td | 1672 "int64_t":$insertSplitDimension,
|