Searched refs:predictedShape (Results 1 – 2 of 2) sorted by relevance
220 ShapedTypeComponents predictedShape = std::get<1>(it); in propagateShapesInRegion() local245 inferredKnowledge.hasRank = predictedShape.hasRank(); in propagateShapesInRegion()246 if (predictedShape.hasRank()) { in propagateShapesInRegion()247 for (auto dim : predictedShape.getDims()) { in propagateShapesInRegion()
59 auto predictedShape = returnedShapes[0]; in createOpAndInfer() local67 inferredKnowledge.hasRank = predictedShape.hasRank(); in createOpAndInfer()68 if (predictedShape.hasRank()) { in createOpAndInfer()69 for (auto dim : predictedShape.getDims()) { in createOpAndInfer()