Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp352 int64_t innerDim = shape.size(); in getContiguousInnerDim() local
356 if (innerDim == 0) in getContiguousInnerDim()
358 const int64_t nextDim = innerDim - 1; in getContiguousInnerDim()
364 innerDim = nextDim; in getContiguousInnerDim()
369 return innerDim; in getContiguousInnerDim()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp121 AffineExpr innerDim = b.getAffineDimExpr(map.getNumDims() - 1); in transferReadSupportsMMAMatrixType() local
123 auto broadcastInnerDim = AffineMap::get(map.getNumDims(), 0, {zero, innerDim}, in transferReadSupportsMMAMatrixType()
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp105 int64_t innerDim = inputShape.back(); in handleMultidimensionalVectors() local
106 int64_t expansionDim = innerDim / vectorWidth; in handleMultidimensionalVectors()
107 assert((innerDim % vectorWidth == 0) && "invalid inner dimension size"); in handleMultidimensionalVectors()