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.