modeci_mdf.interfaces.pytorch.importer.process_torch_schema

modeci_mdf.interfaces.pytorch.importer.process_torch_schema(node: Node, consts: Dict, port_mapper: PortMapper) → Tuple[List[str], Dict[str, Any]]

Parse a TorchScript node schema into argument names and constant attributes (parameters in MDF)

Parameters:

• node – The TorchScript node to retrieve the schema from.

• consts – The constant nodes Dict for the graph we are working with.

Returns:

A tuple containing a list of argument names and Dict of parameter names and values.