| /llvm-project/mlir/include/mlir/Dialect/Mesh/IR/ |
| H A D | MeshOps.h | 112 shardDimension(int64_t dimSize,int64_t shardCount) shardDimension() argument
121 gatherDimension(int64_t dimSize,int64_t shardCount) gatherDimension() argument
|
| /llvm-project/mlir/lib/Dialect/Quant/Utils/ |
| H A D | UniformSupport.cpp | 65 size_t dimSize = type.getDimSize(quantizationDim); in convert() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/Transforms/ |
| H A D | Transforms.cpp | 142 Value dimSize = builder.create<tensor::DimOp>(op.getOperand(), i); in matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Padding.cpp | 42 int64_t dimSize = shape[en.index()]; in computePaddedShape() local
|
| H A D | Vectorization.cpp | 780 auto dimSize = broadcastIfNeeded( calculateGatherOffset() local
803 __anon729377361002(int64_t dimSize) isLoopInvariantIdx() argument
865 __anon729377361202(int64_t dimSize) isContiguousLoadIdx() argument
940 __anon729377361402(int64_t dimSize) getTensorExtractMemoryAccessPattern() argument
|
| /llvm-project/mlir/lib/Dialect/MemRef/Transforms/ |
| H A D | ExpandStridedMetadata.cpp | 285 uint64_t dimSize = expandShapeType.getDimSize(reassocGroup[i]); in getExpandedSizes() local
353 uint64_t dimSize = expandShapeType.getDimSize(reassocGroup[i]); in getExpandedStrides() local
|
| /llvm-project/mlir/lib/Dialect/Affine/Analysis/ |
| H A D | Utils.cpp | 960 int64_t dimSize = memRefType.getDimSize(r); in getConstantBoundingSizeAndShape() local
982 auto dimSize = memRefType.getDimSize(d); in getConstantBoundingSizeAndShape() local
1290 int64_t dimSize = loadOrStoreOp.getMemRefType().getDimSize(r); in boundCheckLoadOrStoreOp() local
|
| /llvm-project/mlir/lib/Conversion/VectorToSCF/ |
| H A D | VectorToSCF.cpp | 1124 int64_t dimSize = xferVecType.getShape()[0]; matchAndRewrite() local
1254 int64_t dimSize = inputVectorTy.getShape()[0]; matchAndRewrite() local
[all...] |
| /llvm-project/mlir/lib/Dialect/ArmSME/Transforms/ |
| H A D | VectorLegalization.cpp | 670 auto dimSize = rewriter.create<arith::ConstantIndexOp>(loc, size); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | LowerVectorContract.cpp | 1041 int64_t dimSize = -1; in lowerParallel() local
1141 int64_t dimSize = lhsType.getDimSize(lhsIndex); in lowerReduction() local
|
| H A D | VectorTransferOpTransforms.cpp | 310 __anon8a39d2070202(int64_t dimSize) getReducedRank() argument
|
| /llvm-project/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 351 int64_t dimSize = memRefType.getDimSize(i); extractSizeOfRankedMemRef() local
1175 Value dimSize; convertSourceMemRefToDescriptor() local
[all...] |
| /llvm-project/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorTilingInterfaceImpl.cpp | 161 OpFoldResult dimSize = srcDimValues[dim]; getTiledImplementation() local
|
| /llvm-project/mlir/lib/Dialect/Mesh/IR/ |
| H A D | MeshOps.cpp | 274 for (int64_t dimSize : getShape()) { verify() local
|
| /llvm-project/mlir/lib/Dialect/Affine/Utils/ |
| H A D | LoopUtils.cpp | 1815 int64_t dimSize = cast<MemRefType>(region.memref.getType()).getDimSize(d); getMultiLevelStrides() local
2257 auto dimSize = memRefType.getDimSize(d); getFullMemRefAsRegion() local
|
| /llvm-project/flang/lib/Evaluate/ |
| H A D | fold-integer.cpp | 253 if (auto &dimSize{shape->at(*dim)}) { UBOUND() local
|
| /llvm-project/mlir/lib/Target/Cpp/ |
| H A D | TranslateToCpp.cpp | 1594 for (auto dimSize : shape) { emitType() local
|
| /llvm-project/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 263 if (!ShapedType::isDynamic(dimSize)) { in matchAndRewrite() local
3477 int64_t dimSize = memrefType.getDimSize(dim); matchAndRewrite() local
|
| /llvm-project/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 113 int64_t dimSize = in getMaskFormat() local
5969 std::optional dimSize = getConstantIntValue(operand); matchAndRewrite() local
[all...] |
| /llvm-project/mlir/lib/IR/ |
| H A D | AsmPrinter.cpp | 1904 __anonc242fc4b1402(const auto &dimSize) printDimensionList() argument
|