Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorMultiDimReductionTransforms.cpp144 int flattenedReductionDim = 0; in matchAndRewrite() local
151 flattenedReductionDim = 1; in matchAndRewrite()
153 flattenedReductionDim *= d; in matchAndRewrite()
156 assert((flattenedParallelDim || flattenedReductionDim) && in matchAndRewrite()
179 if (flattenedReductionDim) { in matchAndRewrite()
181 vectorShape.push_back(flattenedReductionDim); in matchAndRewrite()