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

Class GraphState

exir/serde/export_serialize.py:416–424  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

414
415@dataclass
416class GraphState:
417 inputs: List[Argument] = field(default_factory=list)
418 outputs: List[Argument] = field(default_factory=list)
419 nodes: List[Node] = field(default_factory=list)
420 tensor_values: Dict[str, TensorMeta] = field(default_factory=dict)
421 sym_int_values: Dict[str, SymInt] = field(default_factory=dict)
422 sym_bool_values: Dict[str, SymBool] = field(default_factory=dict)
423 is_single_tensor_return: bool = False
424 custom_obj_values: Dict[str, CustomObjArgument] = field(default_factory=dict)
425
426
427class Final(type):

Callers 2

__init__Method · 0.85
save_graph_stateMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected