Searched defs:SymbolicShapeGraphAnalyzer (Results 1 – 1 of 1) sorted by relevance
804 struct SymbolicShapeGraphAnalyzer { struct805 SymbolicShapeGraphAnalyzer( in SymbolicShapeGraphAnalyzer() argument814 std::optional<ShapeComputeGraphMapping> run() { in run()916 void updateGraphWithSymbolicShapeEqualities( in updateGraphWithSymbolicShapeEqualities()945 void registerStitchedComputeOutput( in registerStitchedComputeOutput()957 void joinPartialEvaluatedShapeGraphToLargeShapeGraph( in joinPartialEvaluatedShapeGraphToLargeShapeGraph()1075 propagateShapesAndGatherPartialEvalShapeGraphs(AliasDb& db) { in propagateShapesAndGatherPartialEvalShapeGraphs()1087 enclosing_graph_value_to_shape_graph_input_;1088 std::unordered_map<int64_t, Value*> symbolic_shape_value_to_graph_output_;1089 std::unordered_map<size_t, int64_t> output_index_to_symbolic_shape_;[all …]