Home
last modified time | relevance | path

Searched refs:inferredKnowledge (Results 1 – 2 of 2) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/
H A DTosaInferShapes.cpp242 auto inferredKnowledge = ValueKnowledge::getPessimisticValueState(); in propagateShapesInRegion() local
243 inferredKnowledge.dtype = in propagateShapesInRegion()
245 inferredKnowledge.hasRank = predictedShape.hasRank(); in propagateShapesInRegion()
248 inferredKnowledge.sizes.push_back(dim); in propagateShapesInRegion()
257 ValueKnowledge::join(currentKnowledge, inferredKnowledge); in propagateShapesInRegion()
H A DTosaDecomposeTransposeConv.cpp64 auto inferredKnowledge = in createOpAndInfer() local
66 inferredKnowledge.dtype = resultTy.cast<ShapedType>().getElementType(); in createOpAndInfer()
67 inferredKnowledge.hasRank = predictedShape.hasRank(); in createOpAndInfer()
70 inferredKnowledge.sizes.push_back(dim); in createOpAndInfer()
76 mlir::tosa::ValueKnowledge::join(currentKnowledge, inferredKnowledge); in createOpAndInfer()