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