modeci_mdf.interfaces.onnx.importer

Code for importing ONNX models into MDF.

Functions

convert_file(input_file)

Simple converter from ONNX to MDF.

find_subgraphs(graph[, graph_dict])

Recurse through an ONNX graph and find all subgraphs.

get_category_of_onnx_node(entry)

get_color_for_onnx_category(shape)

get_onnx_attribute(a)

get_shape_params(shape)

Small helper function to extract a tuple from the TensorShapeProto.

id_to_port(id)

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

main()

onnx_node_to_mdf(node, onnx_initializer)

Construct an MDF node (and function) from an ONNX NodeProto or ValueInfoProto

onnx_to_mdf(onnx_model[, onnx_initializer])

Convert a loaded ONNX model into a MDF model.