Home
last modified time | relevance | path

Searched refs:splitFactor (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp252 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 DLinalgTransformOps.cpp2557 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 DLinalgTransformOps.td1671 "int64_t":$splitFactor,