Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp65 size_t dimSize = type.getDimSize(quantizationDim); in convert() local
66 if (dimSize != scales.size()) { in convert()
70 converters.reserve(dimSize); in convert()
71 for (int i = 0, e = dimSize; i != e; ++i) { in convert()
85 return converters[chunkIndex % dimSize].quantizeFloatToInt(old); in convert()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplit.cpp115 OpFoldResult dimSize = getAsOpFoldResult(iterationSpaceShapes[dimension]); in splitOp() local
119 {splitPoint, dimSize}); in splitOp()
137 rewriter, op.getLoc(), d0 - d1, {dimSize, minSplitPoint}); in splitOp()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp363 int64_t dimSize = memRefType.getDimSize(r); in getConstantBoundingSizeAndShape() local
364 if (ShapedType::isDynamic(dimSize)) in getConstantBoundingSizeAndShape()
366 cstWithShapeBounds.addBound(FlatAffineValueConstraints::UB, r, dimSize - 1); in getConstantBoundingSizeAndShape()
385 auto dimSize = memRefType.getDimSize(d); in getConstantBoundingSizeAndShape() local
386 if (dimSize == -1) in getConstantBoundingSizeAndShape()
388 diffConstant = dimSize; in getConstantBoundingSizeAndShape()
686 int64_t dimSize = loadOrStoreOp.getMemRefType().getDimSize(r); in boundCheckLoadOrStoreOp() local
688 if (dimSize == -1) in boundCheckLoadOrStoreOp()
692 ucst.addBound(FlatAffineValueConstraints::LB, r, dimSize); in boundCheckLoadOrStoreOp()
/llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/
H A DMemRefToLLVM.cpp436 int64_t dimSize = memRefType.getDimSize(i); in extractSizeOfRankedMemRef() local
437 return createIndexConstant(rewriter, loc, dimSize); in extractSizeOfRankedMemRef()
1178 Value dimSize; in convertSourceMemRefToDescriptor() local
1183 dimSize = createIndexConstant(rewriter, loc, size); in convertSourceMemRefToDescriptor()
1187 dimSize = rewriter.create<memref::LoadOp>(loc, shapeOp, index); in convertSourceMemRefToDescriptor()
1189 if (dimSize.getType() != indexType) in convertSourceMemRefToDescriptor()
1190 dimSize = typeConverter->materializeTargetConversion( in convertSourceMemRefToDescriptor()
1191 rewriter, loc, indexType, dimSize); in convertSourceMemRefToDescriptor()
1192 assert(dimSize && "Invalid memref element type"); in convertSourceMemRefToDescriptor()
1195 desc.setSize(rewriter, loc, i, dimSize); in convertSourceMemRefToDescriptor()
[all …]
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToSCF/
H A DVectorToSCF.cpp907 int64_t dimSize = xferVecType.getShape()[0]; in matchAndRewrite() local
911 for (int64_t i = 0; i < dimSize; ++i) { in matchAndRewrite()
1030 int64_t dimSize = xferVecType.getShape()[0]; in matchAndRewrite() local
1036 for (int64_t i = 0; i < dimSize; ++i) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferOpTransforms.cpp247 return llvm::count_if(shape, [](int64_t dimSize) { return dimSize != 1; }); in getReducedRank() argument
H A DVectorTransforms.cpp1873 int64_t dimSize = -1; in lowerParallel() local
1881 dimSize = lhsType.getDimSize(lhsIndex); in lowerParallel()
1884 dimSize = rhsType.getDimSize(rhsIndex); in lowerParallel()
1899 if (resIndex == -1 && dimSize != 1) in lowerParallel()
1916 for (int64_t d = 0; d < dimSize; ++d) { in lowerParallel()
1955 int64_t dimSize = lhsType.getDimSize(lhsIndex); in lowerReduction() local
1956 if (dimSize != rhsType.getDimSize(rhsIndex)) in lowerReduction()
1986 for (int64_t d = 0; d < dimSize; ++d) { in lowerReduction()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp176 int64_t dimSize = memrefType.getDimSize(dim); in matchAndRewrite() local
178 if (dimSize != -1) { in matchAndRewrite()
179 newShapeConstants.push_back(dimSize); in matchAndRewrite()
2786 int64_t dimSize = memrefType.getDimSize(dim); in matchAndRewrite() local
2788 if (!ShapedType::isDynamic(dimSize)) { in matchAndRewrite()
2789 newShapeConstants.push_back(dimSize); in matchAndRewrite()
2799 newShapeConstants.push_back(dimSize); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DLoopUtils.cpp1921 int64_t dimSize = region.memref.getType().cast<MemRefType>().getDimSize(d); in getMultiLevelStrides() local
1922 stride *= dimSize; in getMultiLevelStrides()
1927 if (bufferShape[d] < dimSize && bufferShape[d - 1] > 1) { in getMultiLevelStrides()
2360 auto dimSize = memRefType.getDimSize(d); in getFullMemRefAsRegion() local
2361 assert(dimSize > 0 && "filtered dynamic shapes above"); in getFullMemRefAsRegion()
2363 regionCst->addBound(IntegerPolyhedron::UB, d, dimSize - 1); in getFullMemRefAsRegion()
/llvm-project-15.0.7/mlir/lib/Target/Cpp/
H A DTranslateToCpp.cpp997 for (auto dimSize : shape) { in emitType() local
999 os << dimSize; in emitType()
/llvm-project-15.0.7/flang/lib/Evaluate/
H A Dfold-integer.cpp225 if (auto &dimSize{shape->at(*dim)}) { in UBOUND() local
227 ConvertToType<T>(Expr<ExtentType>{std::move(*dimSize)})); in UBOUND()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp4810 int64_t dimSize = cast<arith::ConstantIndexOp>(defOp).value(); in matchAndRewrite() local
4811 dimSize = std::min(dimSize, maxDimSize); in matchAndRewrite()
4813 if (dimSize <= 0) { in matchAndRewrite()
4817 maskDimSizes.push_back(dimSize); in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.td671 dynamically via `ceilDiv(dimSize[i], num_threads[i])`.