Home
last modified time | relevance | path

Searched defs:shapedType (Results 1 – 25 of 27) sorted by relevance

12

/llvm-project/mlir/lib/CAPI/IR/
H A DBuiltinAttributes.cpp513 MlirAttribute mlirDenseElementsAttrGet(MlirType shapedType, in mlirDenseElementsAttrRawBufferGet() argument
504 mlirDenseElementsAttrGet(MlirType shapedType,intptr_t numElements,MlirAttribute const * elements) mlirDenseElementsAttrGet() argument
526 mlirDenseElementsAttrSplatGet(MlirType shapedType,MlirAttribute element) mlirDenseElementsAttrSplatGet() argument
531 mlirDenseElementsAttrBoolSplatGet(MlirType shapedType,bool element) mlirDenseElementsAttrBoolSplatGet() argument
536 mlirDenseElementsAttrUInt8SplatGet(MlirType shapedType,uint8_t element) mlirDenseElementsAttrUInt8SplatGet() argument
541 mlirDenseElementsAttrInt8SplatGet(MlirType shapedType,int8_t element) mlirDenseElementsAttrInt8SplatGet() argument
546 mlirDenseElementsAttrUInt32SplatGet(MlirType shapedType,uint32_t element) mlirDenseElementsAttrUInt32SplatGet() argument
551 mlirDenseElementsAttrInt32SplatGet(MlirType shapedType,int32_t element) mlirDenseElementsAttrInt32SplatGet() argument
556 mlirDenseElementsAttrUInt64SplatGet(MlirType shapedType,uint64_t element) mlirDenseElementsAttrUInt64SplatGet() argument
561 mlirDenseElementsAttrInt64SplatGet(MlirType shapedType,int64_t element) mlirDenseElementsAttrInt64SplatGet() argument
566 mlirDenseElementsAttrFloatSplatGet(MlirType shapedType,float element) mlirDenseElementsAttrFloatSplatGet() argument
571 mlirDenseElementsAttrDoubleSplatGet(MlirType shapedType,double element) mlirDenseElementsAttrDoubleSplatGet() argument
577 mlirDenseElementsAttrBoolGet(MlirType shapedType,intptr_t numElements,const int * elements) mlirDenseElementsAttrBoolGet() argument
587 getDenseAttribute(MlirType shapedType,intptr_t numElements,const T * elements) getDenseAttribute() argument
594 mlirDenseElementsAttrUInt8Get(MlirType shapedType,intptr_t numElements,const uint8_t * elements) mlirDenseElementsAttrUInt8Get() argument
599 mlirDenseElementsAttrInt8Get(MlirType shapedType,intptr_t numElements,const int8_t * elements) mlirDenseElementsAttrInt8Get() argument
604 mlirDenseElementsAttrUInt16Get(MlirType shapedType,intptr_t numElements,const uint16_t * elements) mlirDenseElementsAttrUInt16Get() argument
609 mlirDenseElementsAttrInt16Get(MlirType shapedType,intptr_t numElements,const int16_t * elements) mlirDenseElementsAttrInt16Get() argument
614 mlirDenseElementsAttrUInt32Get(MlirType shapedType,intptr_t numElements,const uint32_t * elements) mlirDenseElementsAttrUInt32Get() argument
619 mlirDenseElementsAttrInt32Get(MlirType shapedType,intptr_t numElements,const int32_t * elements) mlirDenseElementsAttrInt32Get() argument
624 mlirDenseElementsAttrUInt64Get(MlirType shapedType,intptr_t numElements,const uint64_t * elements) mlirDenseElementsAttrUInt64Get() argument
629 mlirDenseElementsAttrInt64Get(MlirType shapedType,intptr_t numElements,const int64_t * elements) mlirDenseElementsAttrInt64Get() argument
634 mlirDenseElementsAttrFloatGet(MlirType shapedType,intptr_t numElements,const float * elements) mlirDenseElementsAttrFloatGet() argument
639 mlirDenseElementsAttrDoubleGet(MlirType shapedType,intptr_t numElements,const double * elements) mlirDenseElementsAttrDoubleGet() argument
644 mlirDenseElementsAttrBFloat16Get(MlirType shapedType,intptr_t numElements,const uint16_t * elements) mlirDenseElementsAttrBFloat16Get() argument
651 mlirDenseElementsAttrFloat16Get(MlirType shapedType,intptr_t numElements,const uint16_t * elements) mlirDenseElementsAttrFloat16Get() argument
659 mlirDenseElementsAttrStringGet(MlirType shapedType,intptr_t numElements,MlirStringRef * strs) mlirDenseElementsAttrStringGet() argument
672 mlirDenseElementsAttrReshapeGet(MlirAttribute attr,MlirType shapedType) mlirDenseElementsAttrReshapeGet() argument
782 mlirUnmanagedDenseResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,void * data,size_t dataLength,size_t dataAlignment,bool dataIsMutable,void (* deleter)(void * userData,const void * data,size_t size,size_t align),void * userData) mlirUnmanagedDenseResourceElementsAttrGet() argument
802 getDenseResource(MlirType shapedType,MlirStringRef name,intptr_t numElements,const T * elements) getDenseResource() argument
810 mlirUnmanagedDenseBoolResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const int * elements) mlirUnmanagedDenseBoolResourceElementsAttrGet() argument
816 mlirUnmanagedDenseUInt8ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const uint8_t * elements) mlirUnmanagedDenseUInt8ResourceElementsAttrGet() argument
822 mlirUnmanagedDenseUInt16ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const uint16_t * elements) mlirUnmanagedDenseUInt16ResourceElementsAttrGet() argument
828 mlirUnmanagedDenseUInt32ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const uint32_t * elements) mlirUnmanagedDenseUInt32ResourceElementsAttrGet() argument
834 mlirUnmanagedDenseUInt64ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const uint64_t * elements) mlirUnmanagedDenseUInt64ResourceElementsAttrGet() argument
840 mlirUnmanagedDenseInt8ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const int8_t * elements) mlirUnmanagedDenseInt8ResourceElementsAttrGet() argument
846 mlirUnmanagedDenseInt16ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const int16_t * elements) mlirUnmanagedDenseInt16ResourceElementsAttrGet() argument
852 mlirUnmanagedDenseInt32ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const int32_t * elements) mlirUnmanagedDenseInt32ResourceElementsAttrGet() argument
858 mlirUnmanagedDenseInt64ResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const int64_t * elements) mlirUnmanagedDenseInt64ResourceElementsAttrGet() argument
864 mlirUnmanagedDenseFloatResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const float * elements) mlirUnmanagedDenseFloatResourceElementsAttrGet() argument
870 mlirUnmanagedDenseDoubleResourceElementsAttrGet(MlirType shapedType,MlirStringRef name,intptr_t numElements,const double * elements) mlirUnmanagedDenseDoubleResourceElementsAttrGet() argument
936 mlirSparseElementsAttribute(MlirType shapedType,MlirAttribute denseIndices,MlirAttribute denseValues) mlirSparseElementsAttribute() argument
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/IR/
H A DValueBoundsOpInterfaceImpl.cpp40 auto shapedType = llvm::cast<ShapedType>(operand.getType()); in populateBoundsForIndexValue() local
H A DLinalgInterfaces.cpp1100 auto shapedType = llvm::cast<ShapedType>(opOperand.get().getType()); reifyResultShapes() local
H A DLinalgOps.cpp100 auto shapedType = llvm::cast<ShapedType>(source.getType()); createFoldedDimOp() local
/llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h47 if (auto shapedType = dyn_cast<ShapedType>(type)) { in getKnowledgeFromType() local
/llvm-project/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h116 ShapedTypeComponents(ShapedType shapedType) : attr(nullptr) { in ShapedTypeComponents() argument
/llvm-project/mlir/lib/Dialect/Vector/Transforms/
H A DVectorDropLeadUnitDim.cpp234 auto shapedType = cast<ShapedType>(read.getSource().getType()); in matchAndRewrite() local
288 auto shapedType = dyn_cast<ShapedType>(write.getSource().getType()); in matchAndRewrite() local
/llvm-project/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp39 auto shapedType = dyn_cast<ShapedType>(result.getType()); in reifyResultShapes() local
H A DValueBoundsOpInterface.cpp167 } else if (auto shapedType = dyn_cast<ShapedType>(value.getType())) { in assertValidValueDim() local
211 auto shapedType = dyn_cast<ShapedType>(value.getType()); in getExpr() local
376 auto shapedType in processWorklist() local
[all...]
/llvm-project/mlir/lib/Bindings/Python/
H A DIRAttributes.cpp846 shapedType = *explicitType; in getSplat() argument
682 MlirType shapedType; getFromList() local
823 MlirType shapedType; getFromBuffer() local
882 MlirType shapedType = mlirAttributeGetType(*this); accessBuffer() local
1000 bufferInfo(MlirType shapedType,const char * explicitFormat=nullptr) bufferInfo() argument
[all...]
/llvm-project/mlir/lib/IR/
H A DTypeUtilities.cpp145 shapedTypes, [](auto shapedType) { return shapedType.hasRank(); }); in verifyCompatibleShapes() argument
H A DOperation.cpp1142 if (auto shapedType = dyn_cast<ShapedType>(type)) in verifySameOperandsAndResultRank() local
/llvm-project/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp572 auto shapedType = dyn_cast<ShapedType>(valueToTile.getType()); in materializeTiledShape() local
607 auto shapedType = dyn_cast<ShapedType>(valueToTile.getType()); computeSliceParameters() local
/llvm-project/mlir/lib/Dialect/Linalg/TransformOps/
H A DLinalgMatchOps.cpp423 if (auto shapedType = dyn_cast<ShapedType>(type)) { in matchValue() local
/llvm-project/mlir/lib/AsmParser/
H A DAttributeParser.cpp1032 ShapedType shapedType = dyn_cast<ShapedType>(attrType); parseDenseResourceElementsAttr() local
/llvm-project/mlir/lib/Conversion/ArithToSPIRV/
H A DArithToSPIRV.cpp345 if (auto shapedType = dyn_cast<ShapedType>(srcType)) { in matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/Vector/IR/
H A DVectorOps.cpp154 AffineMap mlir::vector::getTransferMinorIdentityMap(ShapedType shapedType, in getTransferMinorIdentityMap() argument
3884 verifyTransferOp(VectorTransferOpInterface op,ShapedType shapedType,VectorType vectorType,VectorType maskType,VectorType inferredMaskType,AffineMap permutationMap,ArrayAttr inBounds) verifyTransferOp() argument
4025 auto shapedType = llvm::dyn_cast<ShapedType>(types[0]); parse() local
4069 ShapedType shapedType = getShapedType(); verify() local
4400 ShapedType shapedType = llvm::dyn_cast<ShapedType>(types[1]); parse() local
4448 ShapedType shapedType = getShapedType(); verify() local
[all...]
/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp2754 bindShapeDims(ShapedType shapedType) bindShapeDims() argument
2757 bindShapeDims(ShapedType shapedType,IntTy & val,IntTy2 &...vals) bindShapeDims() argument
2764 bindShapeDims(ShapedType shapedType,IntTy &...vals) bindShapeDims() argument
/llvm-project/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp136 auto shapedType = llvm::cast<ShapedType>(shapedValue.getType()); in populateDynamicDimSizes() local
/llvm-project/mlir/lib/Target/SPIRV/Serialization/
H A DSerializer.cpp784 auto shapedType = dyn_cast<ShapedType>(valueAttr.getType()); prepareDenseElementsConstant() local
/llvm-project/mlir/lib/Dialect/Arith/IR/
H A DArithOps.cpp151 ShapedType shapedType = llvm::dyn_cast_or_null<ShapedType>(type); getBoolAttribute() local
172 if (auto shapedType = llvm::dyn_cast<ShapedType>(type)) getI1SameShape() local
/llvm-project/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalg.cpp595 auto shapedType = dyn_cast<ShapedType>(tensor.getType()); expandRank() local
656 auto shapedType = dyn_cast<ShapedType>(tensor.getType()); getOrFoldTensorDim() local
/llvm-project/mlir/lib/Dialect/SPIRV/IR/
H A DSPIRVOps.cpp600 auto shapedType = llvm::dyn_cast<ShapedType>(valueType); in verifyConstantType() local
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp207 hasZeroDimension(ShapedType shapedType) hasZeroDimension() argument
/llvm-project/mlir/lib/Dialect/Tensor/IR/
H A DTensorOps.cpp1530 auto shapedType = llvm::dyn_cast<ShapedType>(type); fold() local
2429 foldIdentityOffsetSizeAndStrideOpInterface(OffsetSizeAndStrideOpInterface op,ShapedType shapedType) foldIdentityOffsetSizeAndStrideOpInterface() argument

12