Operator Gen¶
-
systemds.onnx_systemds.operator_gen.
gen_1input_1output_mat_operator
(env: jinja2.environment.Environment, graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto, node: onnx.onnx_ONNX_REL_1_7_ml_pb2.NodeProto) → systemds.onnx_systemds.operator_gen.GeneratedScriptPart¶ Generates simple operators like ‘y = -x’ which have one input and one output. :param env: Jinja environment to load the template files :param graph: the onnx-graph for which the script shall be generated :param node: the onnx-node for which the script shall be generated :return: The generated script part
-
systemds.onnx_systemds.operator_gen.
gen_2input_1output_operator
(env: jinja2.environment.Environment, graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto, node: onnx.onnx_ONNX_REL_1_7_ml_pb2.NodeProto) → systemds.onnx_systemds.operator_gen.GeneratedScriptPart¶ Generates simple operator calls like ‘z = x + y’ which have two inputs (left and right) and one output. :param env: Jinja environment to load the template files :param graph: the onnx-graph for which the script shall be generated :param node: the onnx-node for which the script shall be generated :return: The generated script part
-
systemds.onnx_systemds.operator_gen.
gen_simple_function_call
(env: jinja2.environment.Environment, graph: onnx.onnx_ONNX_REL_1_7_ml_pb2.GraphProto, node: onnx.onnx_ONNX_REL_1_7_ml_pb2.NodeProto) → systemds.onnx_systemds.operator_gen.GeneratedScriptPart¶ Generates a simple function call by directly providing the node inputs as arguments and node outputs as outputs to a function call. Additionally adds the required imports.
- Parameters
env – Jinja environment to load the template files
graph – the onnx-graph for which the script shall be generated
node – the onnx-node for which the script shall be generated
- Returns
The generated script part