Home
last modified time | relevance | path

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

12

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h36 : hasError(false), hasRank(hasRank), dtype(dtype) { in ValueKnowledge()
70 if (hasRank) in getType()
76 return hasRank == rhs.hasRank && sizes == rhs.sizes && dtype == rhs.dtype;
95 if (!lhs.hasRank && !rhs.hasRank) in join()
98 if (!rhs.hasRank) { in join()
99 result.hasRank = true; in join()
104 if (!lhs.hasRank) { in join()
113 result.hasRank = true; in join()
147 if (!lhs.hasRank || !rhs.hasRank) { in meet()
157 result.hasRank = true; in meet()
[all …]
/llvm-project-15.0.7/mlir/lib/Interfaces/
H A DInferTypeOpInterface.cpp25 bool ShapeAdaptor::hasRank() const { in hasRank() function in ShapeAdaptor
29 return t.cast<ShapedType>().hasRank(); in hasRank()
32 return val.get<ShapedTypeComponents *>()->hasRank(); in hasRank()
46 assert(hasRank()); in getDims()
63 assert(hasRank()); in getDims()
69 assert(hasRank()); in getDimSize()
81 assert(hasRank()); in getRank()
90 if (!hasRank()) in hasStaticShape()
132 if (!hasRank()) { in dump()
193 if (shapeAndType.hasRank()) in inferReturnTensorTypes()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp436 if (inputShape.hasRank()) { in inferReturnTypeComponents()
444 if (biasShape.hasRank()) { in inferReturnTypeComponents()
467 if (lhsShape.hasRank()) { in inferReturnTypeComponents()
472 if (rhsShape.hasRank()) { in inferReturnTypeComponents()
493 if (!inputShape.hasRank() && !paddingShape.hasRank()) { in inferReturnTypeComponents()
652 if (!inputShape.hasRank() || !permsShape.hasRank() || in inferReturnTypeComponents()
878 if (!shape.hasRank()) { in REDUCE_SHAPE_INFER()
1044 if (biasShape.hasRank()) { in inferReturnTypeComponents()
1115 if (biasShape.hasRank()) { in inferReturnTypeComponents()
1216 if (biasShape.hasRank()) { in inferReturnTypeComponents()
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp52 if (inferredTy.hasRank()) { in propagateShapesToTosaIf()
87 if (shapedTypeComponent.hasRank()) { in propagateShapesToTosaWhile()
245 inferredKnowledge.hasRank = predictedShape.hasRank(); in propagateShapesInRegion()
246 if (predictedShape.hasRank()) { in propagateShapesInRegion()
269 if (result.hasRank()) { in propagateShapesInRegion()
H A DTosaDecomposeTransposeConv.cpp67 inferredKnowledge.hasRank = predictedShape.hasRank(); in createOpAndInfer()
68 if (predictedShape.hasRank()) { in createOpAndInfer()
H A DTosaDecomposeConv2D.cpp39 if (!weightType.hasRank()) in matchAndRewrite()
/llvm-project-15.0.7/mlir/unittests/Interfaces/
H A DInferTypeOpInterfaceTest.cpp59 EXPECT_TRUE(range.getValueAsShape(1).hasRank()); in TEST_F()
77 EXPECT_TRUE(range.getValueAsShape(0).hasRank()); in TEST_F()
81 EXPECT_TRUE(range.getValueAsShape(1).hasRank()); in TEST_F()
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.h47 bool hasRank() const;
111 ranked = shapedType.hasRank(); in ShapedTypeComponents()
117 ranked = adaptor.hasRank(); in ShapedTypeComponents()
141 bool hasRank() const { return ranked; }; in hasRank() function
/llvm-project-15.0.7/mlir/lib/IR/
H A DTypeUtilities.cpp86 if (!sType1.hasRank() || !sType2.hasRank()) in verifyCompatibleShape()
144 shapedTypes, [](auto shapedType) { return shapedType.hasRank(); })); in verifyCompatibleShapes()
/llvm-project-15.0.7/flang/include/flang/Lower/
H A DSymbolMap.h126 bool hasRank() const { in hasRank() function
130 [](const PointerOrAllocatable &x) { return x.hasRank(); }, in hasRank()
131 [](const Box &x) { return x.hasRank(); }, in hasRank()
/llvm-project-15.0.7/mlir/lib/Dialect/AMDGPU/IR/
H A DAMDGPUDialect.cpp38 if (!bufferType.hasRank()) in verifyRawBufferOp()
/llvm-project-15.0.7/flang/lib/Optimizer/Builder/
H A DBoxValue.cpp76 [](const fir::BoxValue &box) { return box.hasRank(); }, in isArray()
77 [](const fir::MutableBoxValue &box) { return box.hasRank(); }, in isArray()
H A DMutableBox.cpp726 if (box.hasRank()) { in genReallocIfNeeded()
735 if (box.hasRank()) in genReallocIfNeeded()
741 if (box.hasRank()) in genReallocIfNeeded()
844 if (!takeLboundsIfRealloc && box.hasRank()) in finalizeRealloc()
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DBuiltinTypeInterfaces.td78 "bool", "hasRank">,
137 assert($_type.hasRank() && "cannot query rank of unranked shaped type");
157 return $_type.hasRank() &&
H A DBuiltinTypes.h85 bool hasRank() const;
121 bool hasRank() const;
/llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/
H A DDialect.cpp264 return !input.hasRank() || !output.hasRank() || input == output; in areCastCompatible()
H A DLowerToAffineLoops.cpp35 assert(type.hasRank() && "expected only ranked shapes"); in convertTensorToMemRef()
/llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/
H A DDialect.cpp264 return !input.hasRank() || !output.hasRank() || input == output; in areCastCompatible()
/llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/
H A DDialect.cpp264 return !input.hasRank() || !output.hasRank() || input == output; in areCastCompatible()
H A DLowerToAffineLoops.cpp35 assert(type.hasRank() && "expected only ranked shapes"); in convertTensorToMemRef()
/llvm-project-15.0.7/mlir/lib/Conversion/GPUToVulkan/
H A DConvertGPULaunchFuncToVulkanLaunchFunc.cpp59 return memRefType.hasRank() && in isSupportedType()
/llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/
H A DTosaToLinalgNamed.cpp514 if (!firstOperandTy.hasRank() || firstOperandTy.isDynamicDim(0)) { in matchAndRewrite()
518 if (!firstOperandTy.hasRank() || firstOperandTy.isDynamicDim(1)) { in matchAndRewrite()
522 if (!secondOperandTy.hasRank() || secondOperandTy.isDynamicDim(2)) { in matchAndRewrite()
579 if (!inputTy.hasRank() || inputTy.isDynamicDim(0)) { in matchAndRewrite()
583 if (!weightTy.hasRank() || weightTy.isDynamicDim(0)) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/
H A DDialect.cpp291 return !input.hasRank() || !output.hasRank() || input == output; in areCastCompatible()
H A DLowerToAffineLoops.cpp35 assert(type.hasRank() && "expected only ranked shapes"); in convertTensorToMemRef()
/llvm-project-15.0.7/mlir/lib/Conversion/MathToLibm/
H A DMathToLibm.cpp68 if (!vecType.hasRank()) in matchAndRewrite()

12