yobx.translate.reverse_graph_builder#

class yobx.translate.reverse_graph_builder.CustomBuilderEmitter(make_model_function: str = 'make_my_model')[source]#: Custom yobx.translate.builder_emitter.BuilderEmitter.

yobx.translate.reverse_graph_builder.to_graph_builder_code(proto: ModelProto, function_name: str = 'build_model') → str[source]#

Produces a code building a model with yobx.xbuilder.GraphBuilder.

Parameters:

proto – model to convert into a code
function_name – function name

Returns:

str

<<<

import numpy as np
import onnx
import onnx.helper as oh
import onnx.numpy_helper as onh
from yobx.translate.reverse_graph_builder import (
    to_graph_builder_code,
)

TFLOAT = onnx.TensorProto.FLOAT
TINT64 = onnx.TensorProto.INT64

model = oh.make_model(
    oh.make_graph(
        [
            oh.make_node(
                "ConstantOfShape",
                ["shape"],
                ["cst"],
                value=onh.from_array(np.array([0], dtype=np.float32)),
            ),
            oh.make_node(
                "ScatterND",
                ["cst", "indices", "updates"],
                ["Z"],
                reduction="add",
            ),
        ],
        "create_graph",
        [
            oh.make_tensor_value_info("shape", TINT64, [None]),
            oh.make_tensor_value_info("indices", TINT64, [None, None]),
            oh.make_tensor_value_info("updates", TFLOAT, [None, None, None]),
        ],
        [oh.make_tensor_value_info("Z", TFLOAT, [None, None, None])],
    ),
    opset_imports=[
        oh.make_opsetid("", 18),
    ],
    ir_version=9,
)
print(to_graph_builder_code(model))

>>>

    import numpy as np
    import onnx
    import onnx.numpy_helper as onh
    from yobx.xbuilder import GraphBuilder, FunctionOptions
    
    
    
    def create_graph(
        op: "GraphBuilder",
        shape: "INT64[None]",
        indices: "INT64[None, None]",
        updates: "FLOAT[None, None, None]",
    ):
        cst = op.ConstantOfShape(shape, value=onh.from_array(np.array([0.0], dtype=np.float32), name='value'), outputs=['cst'])
        Z = op.ScatterND(cst, indices, updates, reduction='add', outputs=['Z'])
        op.Identity(Z, outputs=["Z"])
        return Z
    
    
    def build_model() -> "ModelProto":
        g = GraphBuilder({'': 18}, ir_version=9)
        g.make_tensor_input("shape", onnx.TensorProto.INT64, (None,))
        g.make_tensor_input("indices", onnx.TensorProto.INT64, (None, None))
        g.make_tensor_input("updates", onnx.TensorProto.FLOAT, (None, None, None))
        create_graph(g.op, "shape", "indices", "updates")
        g.make_tensor_output("Z", onnx.TensorProto.FLOAT, (None, None, None), is_dimension=False, indexed=False)
        model = g.to_onnx()
        return model
    
    
    model = build_model()

yobx.translate.reverse_graph_builder.to_graph_pattern_matching(proto: FunctionProto | GraphProto | ModelProto) → str[source]#

Produces a code matching a pattern.

Parameters:: proto – model to convert into a code
Returns:: str

<<<

import numpy as np
import onnx
import onnx.helper as oh
import onnx.numpy_helper as onh
from yobx.translate.reverse_graph_builder import (
    to_graph_pattern_matching,
)

TFLOAT = onnx.TensorProto.FLOAT
TINT64 = onnx.TensorProto.INT64

model = oh.make_model(
    oh.make_graph(
        [
            oh.make_node(
                "ConstantOfShape",
                ["shape"],
                ["cst"],
                value=onh.from_array(np.array([0], dtype=np.float32)),
            ),
            oh.make_node(
                "ScatterND",
                ["cst", "indices", "updates"],
                ["Z"],
                reduction="add",
            ),
        ],
        "create_graph",
        [
            oh.make_tensor_value_info("shape", TINT64, [None]),
            oh.make_tensor_value_info("indices", TINT64, [None, None]),
            oh.make_tensor_value_info("updates", TFLOAT, [None, None, None]),
        ],
        [oh.make_tensor_value_info("Z", TFLOAT, [None, None, None])],
    ),
    opset_imports=[
        oh.make_opsetid("", 18),
    ],
    ir_version=9,
)
print(to_graph_pattern_matching(model))

>>>

    import inspect
    
    node_1_ScatterND = node
    if node_1_ScatterND.op_type != 'ScatterND' or node_1_ScatterND.domain != '':
        return self.none()
    cst = node_1_ScatterND.input[0]
    indices = node_1_ScatterND.input[1]
    updates = node_1_ScatterND.input[2]
    
    # updates has no predecessor.
    
    # indices has no predecessor.
    
    if g.is_used_more_than_once(cst):
        return self.none(node, inspect.currentframe().f_lineno)
    node_0_ConstantOfShape = g.node_before(cst)
    if node_0_ConstantOfShape is None or node_0_ConstantOfShape.op_type != 'ConstantOfShape' or node_0_ConstantOfShape.domain != '':
        return self.none(node, inspect.currentframe().f_lineno)
    shape = node_0_ConstantOfShape.input[0]
    
    # shape has no predecessor.
    
    # list of nodes
    nodes = [node_0_ConstantOfShape, node_1_ScatterND]

yobx.translate.reverse_graph_builder#

This Page