module onnxrt.onnx_tools¶
Functions¶
function |
truncated documentation |
|---|---|
Finds a node input by its name. |
|
Finds a node by its name. |
|
Inserts a node before one node input. |
Documentation¶
Functions to manipulate ONNX file.
-
mlprodict.onnxrt.onnx_tools.find_node_input_name(node, name)¶ Finds a node input by its name. :param node: onnx node :param name: node name :return: input index
-
mlprodict.onnxrt.onnx_tools.find_node_name(model, name)¶ Finds a node by its name. :param model: onnx graph :param name: node name :return: node pointer
-
mlprodict.onnxrt.onnx_tools.insert_node(model, op_type, node, input_index=0, new_name=None, **attrs)¶ Inserts a node before one node input. :param model: onnx graph :param op_type: :param node: node or node name :param input_index: input index or input name :param attrs: node attributes :return: updated graph
