Home
last modified time | relevance | path

Searched refs:ValueKnowledge (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h33 struct ValueKnowledge { struct
34 ValueKnowledge() = delete;
45 static ValueKnowledge getKnowledgeFromType(Type type) { in getKnowledgeFromType() argument
46 ValueKnowledge result = getPessimisticValueState(); in getKnowledgeFromType()
61 static ValueKnowledge getPessimisticValueState() { in getPessimisticValueState() argument
62 return ValueKnowledge(false, {}, Type()); in getPessimisticValueState()
75 bool operator==(const ValueKnowledge &rhs) const { argument
81 static ValueKnowledge join(const ValueKnowledge &lhs, in join() argument
86 ValueKnowledge result = getPessimisticValueState(); in join()
136 static ValueKnowledge meet(const ValueKnowledge &lhs, in meet() argument
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp59 ValueKnowledge operandKnowledge = ValueKnowledge::getKnowledgeFromType( in propagateShapesToTosaIf()
61 ValueKnowledge blockKnowledge = ValueKnowledge::getKnowledgeFromType( in propagateShapesToTosaIf()
63 ValueKnowledge joinedKnowledge = in propagateShapesToTosaIf()
64 ValueKnowledge::join(operandKnowledge, blockKnowledge); in propagateShapesToTosaIf()
128 llvm::SmallVector<ValueKnowledge> yieldTypeInfo; in propagateShapesToTosaWhile()
130 yieldTypeInfo.push_back(ValueKnowledge::getKnowledgeFromType(ty)); in propagateShapesToTosaWhile()
136 ValueKnowledge::getKnowledgeFromType(it.value().getType()); in propagateShapesToTosaWhile()
138 ValueKnowledge::meet(yieldTypeInfo[it.index()], newKnowledge); in propagateShapesToTosaWhile()
239 ValueKnowledge::getKnowledgeFromType(resultTy); in propagateShapesInRegion()
242 auto inferredKnowledge = ValueKnowledge::getPessimisticValueState(); in propagateShapesInRegion()
[all …]
H A DTosaDecomposeTransposeConv.cpp61 mlir::tosa::ValueKnowledge::getKnowledgeFromType(resultTy); in createOpAndInfer()
65 mlir::tosa::ValueKnowledge::getPessimisticValueState(); in createOpAndInfer()
76 mlir::tosa::ValueKnowledge::join(currentKnowledge, inferredKnowledge); in createOpAndInfer()
/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp1332 llvm::SmallVector<ValueKnowledge> resultKnowledge; in inferReturnTypeComponents()
1336 ValueKnowledge::getKnowledgeFromType(operand.getType())); in inferReturnTypeComponents()
1345 auto meet = ValueKnowledge::meet( in inferReturnTypeComponents()
1347 ValueKnowledge::getKnowledgeFromType(it.value().getType())); in inferReturnTypeComponents()
1354 for (const ValueKnowledge &result : resultKnowledge) { in inferReturnTypeComponents()
1376 llvm::SmallVector<ValueKnowledge> resultKnowledge; in inferReturnTypeComponents()
1380 ValueKnowledge::getKnowledgeFromType(operand.getType())); in inferReturnTypeComponents()
1389 if (auto meet = ValueKnowledge::meet( in inferReturnTypeComponents()
1391 ValueKnowledge::getKnowledgeFromType(it.value().getType()))) { in inferReturnTypeComponents()
1397 for (const ValueKnowledge &result : resultKnowledge) { in inferReturnTypeComponents()