Home
last modified time | relevance | path

Searched refs:kDynamicSize (Results 1 – 25 of 41) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp411 concatDimSize = ShapedType::kDynamicSize; in inferReturnTypeComponents()
434 outShape.resize(2, ShapedType::kDynamicSize); in inferReturnTypeComponents()
465 outShape.resize(3, ShapedType::kDynamicSize); in inferReturnTypeComponents()
595 if (dim != ShapedType::kDynamicSize) in inferReturnTypeComponents()
628 if (val != ShapedType::kDynamicSize) { in inferReturnTypeComponents()
635 if (val == ShapedType::kDynamicSize) in inferReturnTypeComponents()
743 int32_t inHeight = ShapedType::kDynamicSize; in inferReturnTypeComponents()
744 int32_t inWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1020 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
1087 int32_t inputWidth = ShapedType::kDynamicSize; in inferReturnTypeComponents()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Utils/
H A DReshapeOpsUtils.cpp49 while (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
59 if (sourceShape[sourceDim] == ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
60 (currTargetShape != ShapedType::kDynamicSize || in getReassociationIndicesForCollapse()
66 if (currTargetShape == ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
67 sourceShape[sourceDim] != ShapedType::kDynamicSize) in getReassociationIndicesForCollapse()
86 if (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h114 result.sizes.resize(lhs.sizes.size(), ShapedType::kDynamicSize); in join()
119 if (lhsSize == ShapedType::kDynamicSize) { in join()
121 } else if (rhsSize == ShapedType::kDynamicSize) { in join()
158 result.sizes.resize(lhs.sizes.size(), ShapedType::kDynamicSize); in meet()
/llvm-project-15.0.7/mlir/lib/Conversion/LLVMCommon/
H A DPattern.cpp126 assert(count(memRefType.getShape(), ShapedType::kDynamicSize) == in getMemRefDescriptorSizes()
133 sizes.push_back(size == ShapedType::kDynamicSize in getMemRefDescriptorSizes()
149 if (size == ShapedType::kDynamicSize) in getMemRefDescriptorSizes()
150 stride = ShapedType::kDynamicSize; in getMemRefDescriptorSizes()
151 if (stride != ShapedType::kDynamicSize) in getMemRefDescriptorSizes()
156 else if (stride == ShapedType::kDynamicSize) in getMemRefDescriptorSizes()
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DViewLikeInterface.cpp103 return printOperandsOrIntegersListImpl<ShapedType::kDynamicSize>(p, values, in printOperandsOrIntegersSizesList()
157 return parseOperandsOrIntegersImpl<ShapedType::kDynamicSize>(parser, values, in parseOperandsOrIntegersSizesList()
252 return decomposeMixedImpl(b, mixedValues, ShapedType::kDynamicSize); in decomposeMixedSizes()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Shape/IR/
H A DShape.h38 int64_t rank = ShapedType::kDynamicSize);
H A DShapeBase.td95 BuildableType<"::mlir::RankedTensorType::get({ShapedType::kDynamicSize}, "
/llvm-project-15.0.7/mlir/lib/IR/
H A DBuiltinTypeInterfaces.cpp26 constexpr int64_t ShapedType::kDynamicSize; member in ShapedType
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp648 newShape.push_back(ShapedType::kDynamicSize); in matchAndRewrite()
827 size = ShapedType::kDynamicSize; in computeTensorReshapeCollapsedType()
984 ShapedType::kDynamicSize); in inferResultType()
1032 ShapedType::kDynamicSize); in inferCanonicalRankReducedResultType()
1053 ShapedType::kDynamicSize); in build()
1468 ShapedType::kDynamicSize); in build()
1878 staticLow[i] == ShapedType::kDynamicSize || in inferResultType()
1879 staticHigh[i] == ShapedType::kDynamicSize) { in inferResultType()
1928 ShapedType::kDynamicSize); in build()
1930 ShapedType::kDynamicSize); in build()
[all …]
H A DTensorTilingInterfaceImpl.cpp119 staticIndices.push_back(ShapedType::kDynamicSize); in bubbleUpPadSlice()
222 dispatchIndexOpFoldResults(sizes, dynDims, shape, ShapedType::kDynamicSize); in bubbleUpPadSlice()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToTensor/
H A DTosaToTensor.cpp42 if (size != ShapedType::kDynamicSize) in matchAndRewrite()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypeInterfaces.td57 dimensions of the shape must be positive, or kDynamicSize (in which case the
90 static constexpr int64_t kDynamicSize = -1;
96 return dSize == kDynamicSize;
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Utils/
H A DUtils.cpp65 if (en.value() == ShapedType::kDynamicSize) in createDynamicDimValues()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaDecomposeConv2D.cpp57 combined = ShapedType::kDynamicSize; in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgTransformOps.cpp788 parser.getBuilder().getI64IntegerAttr(ShapedType::kDynamicSize); in parse()
804 if (staticSplitSize != ShapedType::kDynamicSize) in print()
815 ShapedType::kDynamicSize) ^ in verify()
951 if (size == ShapedType::kDynamicSize) { in getMixedSizes()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorConversion.cpp125 uint64_t s = shape[i] == ShapedType::kDynamicSize ? 0 : shape[i]; in sizesFromType()
146 if (shape[i] == ShapedType::kDynamicSize) in sizesFromPtr()
156 auto memTp = MemRefType::get({ShapedType::kDynamicSize}, tp); in genAlloca()
165 auto memTp = MemRefType::get({ShapedType::kDynamicSize}, tp); in genAlloc()
335 if (shape[i] == ShapedType::kDynamicSize) in allocDenseTensor()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp99 if (reductionDimSize == ShapedType::kDynamicSize || in splitReduction()
292 if (reductionDimSize == ShapedType::kDynamicSize || in splitReductionByScaling()
H A DFusion.cpp159 SmallVector<int64_t, 4> staticSizesVector(rank, ShapedType::kDynamicSize); in fuse()
H A DPromotion.cpp96 ShapedType::kDynamicSize); in defaultAllocBufferCallBack()
H A DHoistPadding.cpp428 SmallVector<int64_t> packedShape(nPackedLoops, ShapedType::kDynamicSize); in hoistPaddingOnTensors()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DPipelineDataTransfer.cpp88 if (dim.value() == ShapedType::kDynamicSize) in doubleBuffer()
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp222 if (shape1[d] != shape2[d] && shape2[d] != ShapedType::kDynamicSize) in verify()
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp47 int64_t asSize() { return saturated ? ShapedType::kDynamicSize : v; } in asSize()
1478 ShapedType::kDynamicSize); in build()
2052 if (shapeSize == ShapedType::kDynamicSize) in verify()
2144 ShapedType::kDynamicSize); in inferResultType()
2189 ShapedType::kDynamicSize); in inferRankReducedResultType()
2210 ShapedType::kDynamicSize); in build()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgOps.td114 shape.size(), ShapedType::kDynamicSize);
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DBuiltinTypes.cpp152 int64_t mlirShapedTypeGetDynamicSize() { return ShapedType::kDynamicSize; } in mlirShapedTypeGetDynamicSize()

12