Home
last modified time | relevance | path

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

/llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/
H A DShapeUtils.h33 struct ValueKnowledge { struct
34 ValueKnowledge() = delete;
35 ValueKnowledge(bool hasRank, llvm::ArrayRef<int64_t> newSizes, Type dtype) in ValueKnowledge() function
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
82 const ValueKnowledge &rhs) { in join()
[all …]
/llvm-project/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp157 ValueKnowledge operandKnowledge = ValueKnowledge::getKnowledgeFromType( in propagateShapesToTosaWhile()
159 ValueKnowledge blockKnowledge = ValueKnowledge::getKnowledgeFromType( in propagateShapesToTosaWhile()
161 ValueKnowledge joinedKnowledge = in propagateShapesToTosaWhile()
162 ValueKnowledge::join(operandKnowledge, blockKnowledge); in propagateShapesToTosaWhile()
204 llvm::SmallVector<ValueKnowledge> yieldTypeInfo; in propagateShapesToTosaWhile()
206 yieldTypeInfo.push_back(ValueKnowledge::getKnowledgeFromType(ty)); in propagateShapesToTosaWhile()
212 ValueKnowledge::getKnowledgeFromType(it.value().getType()); in propagateShapesToTosaWhile()
214 ValueKnowledge in propagateShapesToTosaWhile()
[all...]
H A DTosaDecomposeTransposeConv.cpp
/llvm-project/mlir/lib/Dialect/Tosa/IR/
H A DTosaOps.cpp2048 llvm::SmallVector<ValueKnowledge> resultKnowledge;
2052 ValueKnowledge::getKnowledgeFromType(operand.getType()));
2061 auto meet = ValueKnowledge::meet(
2063 ValueKnowledge::getKnowledgeFromType(it.value().getType()));
2070 for (const ValueKnowledge &result : resultKnowledge) {
2092 llvm::SmallVector<ValueKnowledge> resultKnowledge;
2096 ValueKnowledge::getKnowledgeFromType(operand.getType()));
2105 if (auto meet = ValueKnowledge::meet(
2107 ValueKnowledge::getKnowledgeFromType(it.value().getType()))) {
2113 for (const ValueKnowledge
[all...]