Searched refs:GraphShapeInfo (Results 1 – 6 of 6) sorted by relevance
38 typedef std::unordered_map<string, std::vector<InferredShape>> GraphShapeInfo; typedef45 GraphShapeInfo* shape_info);
49 GraphShapeInfo shape_info; in TEST()113 GraphShapeInfo shape_info; in TEST()
99 GraphShapeInfo* shape_info) { in StoreOutputShapes()137 GraphShapeInfo* shape_info) { in InferShapes()
38 const Graph& graph, const GraphShapeInfo& shape_info,
24 const Graph& graph, const GraphShapeInfo& shape_info, in ShapeAnnotationsMatch()
316 GraphShapeInfo shape_info; in PerformStaticShapeInferenceBeforeEncapsulation()