| 207 | |
| 208 | @dataclass |
| 209 | class Graph: |
| 210 | inputs: List[Argument] |
| 211 | outputs: List[Argument] |
| 212 | nodes: List[Node] |
| 213 | tensor_values: Dict[str, TensorMeta] |
| 214 | sym_int_values: Dict[str, SymInt] |
| 215 | sym_bool_values: Dict[str, SymBool] |
| 216 | # This is for deserializing the submodule graphs from higher order ops |
| 217 | # (ex. cond, map) where single tensor returns will just return a single |
| 218 | # tensor, rather than following export schema and returning a singleton |
| 219 | # list. |
| 220 | is_single_tensor_return: bool = False |
| 221 | custom_obj_values: Dict[str, CustomObjArgument] = field(default_factory=dict) |
| 222 | |
| 223 | |
| 224 | @dataclass |
no outgoing calls