MCPcopy Create free account
hub / github.com/pytorch/executorch / Graph

Class Graph

exir/serde/schema.py:209–221  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

207
208@dataclass
209class 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

Callers 3

test_filter_nodesMethod · 0.85
serialize_graphMethod · 0.85
_canonicalize_graphFunction · 0.85

Calls

no outgoing calls

Tested by 1

test_filter_nodesMethod · 0.68