Home
last modified time | relevance | path

Searched refs:shapedType (Results 1 – 20 of 20) sorted by relevance

/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DBuiltinAttributes.cpp334 MlirAttribute mlirDenseElementsAttrGet(MlirType shapedType, in mlirDenseElementsAttrGet() argument
346 auto shapedTypeCpp = unwrap(shapedType).cast<ShapedType>(); in mlirDenseElementsAttrRawBufferGet()
417 static MlirAttribute getDenseAttribute(MlirType shapedType, in getDenseAttribute() argument
428 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrUInt8Get()
433 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrInt8Get()
438 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrUInt16Get()
443 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrInt16Get()
448 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrUInt32Get()
453 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrInt32Get()
458 return getDenseAttribute(shapedType, numElements, elements); in mlirDenseElementsAttrUInt64Get()
[all …]
/llvm-project-15.0.7/mlir/include/mlir-c/
H A DBuiltinAttributes.h332 MlirType shapedType, size_t rawBufferSize, const void *rawBuffer);
339 mlirDenseElementsAttrBoolSplatGet(MlirType shapedType, bool element);
343 mlirDenseElementsAttrInt8SplatGet(MlirType shapedType, int8_t element);
353 mlirDenseElementsAttrFloatSplatGet(MlirType shapedType, float element);
361 MlirType shapedType, intptr_t numElements, const int *elements);
363 MlirType shapedType, intptr_t numElements, const uint8_t *elements);
365 MlirType shapedType, intptr_t numElements, const int8_t *elements);
369 MlirType shapedType, intptr_t numElements, const int16_t *elements);
379 MlirType shapedType, intptr_t numElements, const float *elements);
381 MlirType shapedType, intptr_t numElements, const double *elements);
[all …]
/llvm-project-15.0.7/mlir/lib/Bindings/Python/
H A DIRAttributes.cpp487 auto shapedType = mlirRankedTensorTypeGet( in getFromBuffer() local
516 if (!mlirTypeIsAShaped(shapedType) || in getSplat()
555 return bufferInfo<float>(shapedType); in accessBuffer()
559 return bufferInfo<double>(shapedType); in accessBuffer()
570 return bufferInfo<int32_t>(shapedType); in accessBuffer()
574 return bufferInfo<uint32_t>(shapedType); in accessBuffer()
581 return bufferInfo<int64_t>(shapedType); in accessBuffer()
585 return bufferInfo<uint64_t>(shapedType); in accessBuffer()
592 return bufferInfo<int8_t>(shapedType); in accessBuffer()
596 return bufferInfo<uint8_t>(shapedType); in accessBuffer()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h47 if (auto shapedType = type.dyn_cast<ShapedType>()) { in getKnowledgeFromType() local
48 if (shapedType.hasRank()) { in getKnowledgeFromType()
50 result.sizes.reserve(shapedType.getRank()); in getKnowledgeFromType()
51 for (auto dim : shapedType.getShape()) in getKnowledgeFromType()
54 result.dtype = shapedType.getElementType(); in getKnowledgeFromType()
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h110 ShapedTypeComponents(ShapedType shapedType) : attr(nullptr) { in ShapedTypeComponents() argument
111 ranked = shapedType.hasRank(); in ShapedTypeComponents()
112 elementType = shapedType.getElementType(); in ShapedTypeComponents()
114 dims = llvm::to_vector<4>(shapedType.getShape()); in ShapedTypeComponents()
/llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/IR/
H A DArithmeticOps.cpp881 if (auto shapedType = resType.dyn_cast<ShapedType>()) in fold() local
882 bitWidth = shapedType.getElementTypeBitWidth(); in fold()
910 if (auto shapedType = resType.dyn_cast<ShapedType>()) in fold() local
911 bitWidth = shapedType.getElementTypeBitWidth(); in fold()
965 bitWidth = shapedType.getElementTypeBitWidth(); in fold()
1061 resEleType = shapedType.getElementType(); in fold()
1087 resEleType = shapedType.getElementType(); in fold()
1112 resEleType = shapedType.getElementType(); in fold()
1138 resEleType = shapedType.getElementType(); in fold()
1306 if (!shapedType) in getBoolAttribute()
[all …]
/llvm-project-15.0.7/mlir/lib/IR/
H A DTypeUtilities.cpp144 shapedTypes, [](auto shapedType) { return shapedType.hasRank(); })); in verifyCompatibleShapes() argument
H A DBuiltinAttributes.cpp934 auto shapedType = denseArrayAttrEltTypeBuilder<T>::getShapedType( in get() local
939 return Base::get(context, shapedType, eltType, rawArray) in get()
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp135 auto shapedType = shapedValue.getType().cast<ShapedType>(); in populateDynamicDimSizes() local
136 for (int64_t i = 0; i < shapedType.getRank(); ++i) { in populateDynamicDimSizes()
137 if (shapedType.isDynamicDim(i)) { in populateDynamicDimSizes()
138 if (shapedType.isa<MemRefType>()) { in populateDynamicDimSizes()
141 assert(shapedType.isa<RankedTensorType>() && "expected tensor"); in populateDynamicDimSizes()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp200 auto shapedType = read.getSource().getType().cast<ShapedType>(); in matchAndRewrite() local
201 if (shapedType.getElementType() != read.getVectorType().getElementType()) in matchAndRewrite()
248 auto shapedType = write.getSource().getType().dyn_cast<ShapedType>(); in matchAndRewrite() local
249 if (shapedType.getElementType() != write.getVectorType().getElementType()) in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp126 shapedType.getElementType().dyn_cast<VectorType>(); in getTransferMinorIdentityMap()
131 if (shapedType.getRank() == 0 && in getTransferMinorIdentityMap()
135 getAffineConstantExpr(0, shapedType.getContext())); in getTransferMinorIdentityMap()
138 shapedType.getContext()); in getTransferMinorIdentityMap()
2859 if (!shapedType.isa<MemRefType, RankedTensorType>()) in verifyTransferOp()
2863 auto elementType = shapedType.getElementType(); in verifyTransferOp()
2988 auto shapedType = types[0].dyn_cast<ShapedType>(); in parse() local
2989 if (!shapedType || !shapedType.isa<MemRefType, RankedTensorType>()) in parse()
3027 ShapedType shapedType = getShapedType(); in verify() local
3448 if (!shapedType || !shapedType.isa<MemRefType, RankedTensorType>()) in parse()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp207 auto shapedType = val.getType().cast<ShapedType>(); in getDynOperands() local
208 for (const auto &dim : llvm::enumerate(shapedType.getShape())) { in getDynOperands()
803 auto shapedType = valueToTile.getType().dyn_cast<ShapedType>(); in makeTiledShape() local
804 assert(shapedType && "only shaped types can be tiled"); in makeTiledShape()
805 ArrayRef<int64_t> shape = shapedType.getShape(); in makeTiledShape()
806 int64_t rank = shapedType.getRank(); in makeTiledShape()
901 auto *sliceOp = TypeSwitch<ShapedType, Operation *>(shapedType) in makeTiledShape()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/IR/
H A DVectorOps.h164 AffineMap getTransferMinorIdentityMap(ShapedType shapedType,
/llvm-project-15.0.7/mlir/lib/Target/SPIRV/Serialization/
H A DSerializer.cpp707 auto shapedType = valueAttr.getType().dyn_cast<ShapedType>(); in prepareDenseElementsConstant() local
708 assert(dim <= shapedType.getRank()); in prepareDenseElementsConstant()
709 if (shapedType.getRank() == dim) { in prepareDenseElementsConstant()
729 operands.reserve(shapedType.getDimSize(dim) + 2); in prepareDenseElementsConstant()
731 for (int i = 0; i < shapedType.getDimSize(dim); ++i) { in prepareDenseElementsConstant()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp1285 static void bindShapeDims(ShapedType shapedType) {} in bindShapeDims() argument
1288 static void bindShapeDims(ShapedType shapedType, IntTy &val, IntTy2 &...vals) { in bindShapeDims() argument
1289 val = shapedType.getShape()[N]; in bindShapeDims()
1290 bindShapeDims<N + 1, IntTy2 &...>(shapedType, vals...); in bindShapeDims()
1295 static void bindShapeDims(ShapedType shapedType, IntTy &...vals) { in bindShapeDims() argument
1296 bindShapeDims<0>(shapedType, vals...); in bindShapeDims()
/llvm-project-15.0.7/mlir/lib/Conversion/ArithmeticToSPIRV/
H A DArithmeticToSPIRV.cpp389 if (auto shapedType = srcType.dyn_cast<ShapedType>()) { in matchAndRewrite() local
390 if (shapedType.getNumElements() != 1) in matchAndRewrite()
392 srcType = shapedType.getElementType(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp743 auto shapedType = type.dyn_cast<ShapedType>(); in fold() local
744 if (shapedType && shapedType.hasRank()) in fold()
745 return IntegerAttr::get(IndexType::get(getContext()), shapedType.getRank()); in fold()
1385 ShapedType shapedType) { in foldIdentityOffsetSizeAndStrideOpInterface() argument
1392 auto shape = shapedType.getShape(); in foldIdentityOffsetSizeAndStrideOpInterface()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgInterfaces.td642 if (auto shapedType =
644 return shapedType.getRank();
670 if (auto shapedType =
672 return shapedType.getShape();
/llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/
H A DMemRefOps.cpp1455 auto shapedType = type.dyn_cast<ShapedType>(); in fold() local
1456 if (shapedType && shapedType.hasRank()) in fold()
1457 return IntegerAttr::get(IndexType::get(getContext()), shapedType.getRank()); in fold()
/llvm-project-15.0.7/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVOps.cpp1836 auto shapedType = valueType.dyn_cast<ShapedType>(); in verifyConstantType() local
1851 auto valueElemType = shapedType.getElementType(); in verifyConstantType()
1858 if (numElements != shapedType.getNumElements()) { in verifyConstantType()
1861 << shapedType.getNumElements() << ")"; in verifyConstantType()