modeci_mdf.interfaces.pytorch.importer.process_onnx_schema

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

Retrieve the argument names and attributes (parameters in MDF) for this Operation.

Parameters:

  • op – The TorchScript node containing the ONNX operation.

  • port_mapper – The utitlity class for assigning TorchScript input output ids to Input Output Port ids.

Returns:

  • A dict representing argument names mapping to input port ids

  • A dict mapping parameters (ONNX attributes) names mapping to values

Return type:

A two element tuple