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