Searched refs:splitFactor (Results 1 – 3 of 3) sorted by relevance
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | SplitReduction.cpp | 252 int64_t splitFactor = control.ratio; in splitReductionByScaling() local
254 if (splitFactor <= 1) in splitReductionByScaling()
266 reductionDimSize % splitFactor != 0 || in splitReductionByScaling()
315 reductionDimSize / splitFactor, insertSplitDimension); in splitReductionByScaling()
338 newMaps.push_back(scaleReductionDim(op, *o, reductionDimPos, splitFactor)); in splitReductionByScaling()
350 reductionDimSize / splitFactor)); in splitReductionByScaling()
358 loc, ArrayRef<int64_t>{reductionDimSize / splitFactor, splitFactor}, in splitReductionByScaling()
|
| /llvm-project/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 2557 int64_t splitFactor, int64_t insertSplitDimension, bool innerParallel, in applyToOne()
2562 builder.getI64IntegerAttr(splitFactor)); 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 | 1671 "int64_t":$splitFactor,
|