Home
last modified time | relevance | path

Searched refs:getDimSize (Results 1 – 25 of 49) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp437 outShape[0] = inputShape.getDimSize(0); in inferReturnTypeComponents()
468 outShape[0] = lhsShape.getDimSize(0); in inferReturnTypeComponents()
469 outShape[1] = lhsShape.getDimSize(1); in inferReturnTypeComponents()
476 outShape[2] = rhsShape.getDimSize(2); in inferReturnTypeComponents()
594 int dim = inputShape.getDimSize(i); in inferReturnTypeComponents()
682 if (inputShape.getDimSize(0) != inputShape.getDimSize(i)) { in inferReturnTypeComponents()
702 outputShape[i] = inputShape.getDimSize(permShape.getDimSize(i)); in inferReturnTypeComponents()
751 inHeight = inputShape.getDimSize(1); in inferReturnTypeComponents()
752 inWidth = inputShape.getDimSize(2); in inferReturnTypeComponents()
892 auto dim2 = shape.getDimSize(i); in REDUCE_SHAPE_INFER()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/AMX/IR/
H A DAMXDialect.cpp34 unsigned col = tp.getDimSize(1) * tp.getElementType().getIntOrFloatBitWidth(); in verifyTileSize()
35 if (tp.getDimSize(0) > kMaxRows) in verifyTileSize()
36 return op->emitOpError("bad row height: ") << tp.getDimSize(0); in verifyTileSize()
46 unsigned am = atp.getDimSize(0), ak = atp.getDimSize(1) >> scale; in verifyMultShape()
47 unsigned bk = btp.getDimSize(0), bn = btp.getDimSize(1) >> scale; in verifyMultShape()
48 unsigned cm = ctp.getDimSize(0), cn = ctp.getDimSize(1); in verifyMultShape()
/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp61 EXPECT_EQ(range.getValueAsShape(1).getDimSize(0), 10); in TEST_F()
63 EXPECT_EQ(range.getShape(1).getDimSize(0), 1); in TEST_F()
79 EXPECT_EQ(range.getValueAsShape(0).getDimSize(0), 30); in TEST_F()
83 EXPECT_EQ(range.getValueAsShape(1).getDimSize(0), 10); in TEST_F()
98 EXPECT_EQ(range.getShape(0).getDimSize(0), 10); in TEST_F()
99 EXPECT_EQ(range.getShape(0).getDimSize(1), 20); in TEST_F()
102 EXPECT_EQ(range.getShape(1).getDimSize(0), 1); in TEST_F()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DNamedOpConversions.cpp47 if (kernelTy.getDimSize(3) != 1) in matchAndReplaceDepthwiseConv()
54 {kernelTy.getDimSize(0), kernelTy.getDimSize(1), kernelTy.getDimSize(2)}, in matchAndReplaceDepthwiseConv()
64 RankedTensorType::get({initTy.getDimSize(0), initTy.getDimSize(1), in matchAndReplaceDepthwiseConv()
65 initTy.getDimSize(2), initTy.getDimSize(3)}, in matchAndReplaceDepthwiseConv()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeTransposeConv.cpp114 int64_t kernelHeight = weightTy.getDimSize(1); in matchAndRewrite()
115 int64_t kernelWidth = weightTy.getDimSize(2); in matchAndRewrite()
185 int64_t batch = inputTy.getDimSize(0); in matchAndRewrite()
187 int64_t outputChannels = weightTy.getDimSize(0); in matchAndRewrite()
188 int64_t weightHeight = weightTy.getDimSize(1); in matchAndRewrite()
189 int64_t weightWidth = weightTy.getDimSize(2); in matchAndRewrite()
190 int64_t inputChannels = weightTy.getDimSize(3); in matchAndRewrite()
217 weightHeight = weightTy.getDimSize(1); in matchAndRewrite()
218 weightWidth = weightTy.getDimSize(2); in matchAndRewrite()
314 int64_t convHeight = convTy.getDimSize(1); in matchAndRewrite()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp2370 if (vLHS.getDimSize(0) != vRES.getDimSize(0)) in verify()
2372 if (vRHS.getDimSize(0) != vRES.getDimSize(1)) in verify()
2378 if (vLHS.getDimSize(0) != vRES.getDimSize(0)) in verify()
3970 if (resVType.getDimSize(0) != maskVType.getDimSize(0)) in verify()
4023 if (valueVType.getDimSize(0) != maskVType.getDimSize(0)) in verify()
4074 if (resVType.getDimSize(0) != indVType.getDimSize(0)) in verify()
4076 if (resVType.getDimSize(0) != maskVType.getDimSize(0)) in verify()
4122 if (valueVType.getDimSize(0) != indVType.getDimSize(0)) in verify()
4124 if (valueVType.getDimSize(0) != maskVType.getDimSize(0)) in verify()
4168 if (resVType.getDimSize(0) != maskVType.getDimSize(0)) in verify()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/IR/
H A DQuantOps.cpp107 if (layerStatsType.getRank() != 1 || layerStatsType.getDimSize(0) != 2) { in verify()
125 if (axisStatsType.getRank() != 2 || axisStatsType.getDimSize(1) != 2 || in verify()
126 axisStatsType.getDimSize(0) != argSliceSize) { in verify()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransforms.cpp310 if (srcType.getDimSize(r) != dstType.getDimSize(r)) { in matchAndRewrite()
804 int64_t dim = dstType.getDimSize(0); in matchAndRewrite()
925 numElts *= sourceVectorType.getDimSize(r); in matchAndRewrite()
951 if (++idx[r] == tp.getDimSize(r)) { in incIdx()
1405 int64_t lhsRows = lhsType.getDimSize(0); in matchAndRewrite()
1406 int64_t lhsColumns = lhsType.getDimSize(1); in matchAndRewrite()
1407 int64_t rhsColumns = rhsType.getDimSize(1); in matchAndRewrite()
1881 dimSize = lhsType.getDimSize(lhsIndex); in lowerParallel()
1884 dimSize = rhsType.getDimSize(rhsIndex); in lowerParallel()
1955 int64_t dimSize = lhsType.getDimSize(lhsIndex); in lowerReduction()
[all …]
H A DVectorUnrollDistribute.cpp588 map[affineMap.getDimPosition(i)] = extract.getResultType().getDimSize(i); in matchAndRewrite()
666 extract.getResultType().getDimSize(vectorPos), read.getContext()); in matchAndRewrite()
713 insert.getSourceVectorType().getDimSize(vectorPos), in matchAndRewrite()
H A DVectorDistribute.cpp255 if (srcType.getDimSize(i) != dstType.getDimSize(i)) in calculateImplicitMap()
610 distributedVal.getType().cast<VectorType>().getDimSize(vectorPos); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/NVGPU/Transforms/
H A DOptimizeSharedMemory.cpp49 1, kSharedMemoryLineSizeBytes / ((memrefTy.getDimSize(tgtDim) * in permuteVectorOffset()
64 int64_t M = llvm::Log2_64(memrefTy.getDimSize(tgtDim)); in permuteVectorOffset()
189 const int64_t rowSize = memRefType.getDimSize(memRefType.getRank() - 1); in optimizeSharedMemoryReadsAndWrites()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/Transforms/
H A DExpandOps.cpp89 int64_t rank = shapeType.cast<MemRefType>().getDimSize(0); in matchAndRewrite()
107 sizes[i] = rewriter.getIndexAttr(op.getType().getDimSize(i)); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DVectorInterfaces.cpp20 shape.push_back(vecType.getDimSize(i)); in transferMaskType()
H A DInferTypeOpInterface.cpp68 int64_t ShapeAdaptor::getDimSize(int index) const { in getDimSize() function in ShapeAdaptor
71 return t.cast<ShapedType>().getDimSize(index); in getDimSize()
/llvm-project-15.0.7/mlir/lib/Dialect/AMX/Transforms/
H A DLegalizeForLLVMExport.cpp33 auto mattr = rewriter.getI16IntegerAttr(vType.getDimSize(0)); in getTileSizes()
34 auto nattr = rewriter.getI16IntegerAttr(vType.getDimSize(1) * bytes); in getTileSizes()
72 auto attr = rewriter.getI64IntegerAttr(mType.getDimSize(last) * bytes); in getStride()
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h62 int64_t getDimSize(int index) const;
67 return ShapedType::isDynamic(getDimSize(index)); in isDynamicDim()
/llvm-project-15.0.7/mlir/lib/Dialect/Quant/Utils/
H A DUniformSupport.cpp65 size_t dimSize = type.getDimSize(quantizationDim); in convert()
/llvm-project-15.0.7/mlir/lib/Conversion/TensorToSPIRV/
H A DTensorToSPIRV.cpp63 strides[i] = strides[i + 1] * tensorType.getDimSize(i + 1); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/NVGPUToNVVM/
H A DNVGPUToNVVM.cpp212 vectorResultType.getElementType(), vectorResultType.getDimSize(1)); in matchAndRewrite()
214 int64_t num32BitRegs = vectorResultType.getDimSize(0); in matchAndRewrite()
240 for (int64_t i = 0, e = vectorResultType.getDimSize(0); i < e; i++) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp177 join.push_back(two.getDimSize(i)); in joinShapes()
181 join.push_back(one.getDimSize(i)); in joinShapes()
184 if (one.getDimSize(i) != two.getDimSize(i)) in joinShapes()
186 join.push_back(one.getDimSize(i)); in joinShapes()
439 stride *= tensorType.getDimSize(i); in fold()
566 getLoc(), getType().getDimSize(dim)); in reifyResultShapes()
769 getShape().getType().cast<RankedTensorType>().getDimSize(0); in verify()
1830 if (resultType.getDimSize(i) == expectedType.getDimSize(i)) in verify()
1883 int64_t size = sourceType.getDimSize(i) + staticLow[i] + staticHigh[i]; in inferResultType()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToLLVM/
H A DConvertVectorToLLVM.cpp109 auto ptrsType = LLVM::getFixedVectorType(pType, vType.getDimSize(0)); in getIndexedPtrs()
571 int64_t v1Dim = v1Type.getDimSize(0); in matchAndRewrite()
1149 int64_t dim = rank == 0 ? 1 : vectorType.getDimSize(0); in emitRanks()
1164 int64_t dim = vectorType.getDimSize(0); in emitRanks()
1218 int64_t width = splatOp.getType().cast<VectorType>().getDimSize(0); in matchAndRewrite()
1264 int64_t width = resultType.getDimSize(resultType.getRank() - 1); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DUtils.cpp363 int64_t dimSize = memRefType.getDimSize(r); in getConstantBoundingSizeAndShape()
385 auto dimSize = memRefType.getDimSize(d); in getConstantBoundingSizeAndShape()
592 memRefType.getDimSize(r) - 1); in compute()
653 sizeInBytes = sizeInBytes * memRefType.getDimSize(i); in getMemRefSizeInBytes()
686 int64_t dimSize = loadOrStoreOp.getMemRefType().getDimSize(r); in boundCheckLoadOrStoreOp()
/llvm-project-15.0.7/mlir/lib/IR/
H A DTypeUtilities.cpp159 [&](auto shape) { return shape.getDimSize(i); })); in verifyCompatibleShapes()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypeInterfaces.td175 int64_t getDimSize(unsigned idx) const {
184 assert(::mlir::ShapedType::isDynamic(getDimSize(index)) && "invalid index");
/llvm-project-15.0.7/mlir/lib/Conversion/LinalgToSPIRV/
H A DLinalgToSPIRV.cpp126 if ((*localSize.begin()).getSExtValue() != originalInputType.getDimSize(0)) in matchAndRewrite()

12