modeci_mdf.interfaces.pytorch.importer.PortMapper

class modeci_mdf.interfaces.pytorch.importer.PortMapper(graph: Graph, args: Tuple)[source]

Bases: object

A simple class that handles mapping TorchScript inputouput ids to MDF InputPortOutputPort ids. It keeps track of annoying details like graph level inputs and stuff.

Methods

id_to_port(id)

Turn unique TorchScript output and input value names into valid MDF input and outport names

port_to_id(name)

Transform a port name back to is TorchScript ID

id_to_port(id: str)[source]

Turn unique TorchScript output and input value names into valid MDF input and outport names

port_to_id(name: str)[source]

Transform a port name back to is TorchScript ID