modeci_mdf.interfaces.pytorch.importer.get_graph_constants
- modeci_mdf.interfaces.pytorch.importer.get_graph_constants(graph: Graph) Dict[str, Any] [source]
Find all constant nodes in the graph and extract their values as a proper JSON serializable value.
- Parameters:
graph – The graph to extract constants from.
- Returns:
A Dict that maps the constant nodes unique TorchScript node ID string to its value.