module onnx_grammar.node_visitor_translator

Short summary

module mlprodict.onnx_grammar.node_visitor_translator

One class which visits a syntax tree.

source on GitHub

Classes

class	truncated documentation
CodeNodeVisitor	Defines a visitor which walks though the syntax tree of the code.

Properties

property	truncated documentation
Rows	returns a list of dictionaries with all the elements of the code

Static Methods

staticmethod	truncated documentation
print_node	Debugging purpose.

Methods

method	truncated documentation
__init__
export	Calls method export from the translator class.
generic_visit	Overrides generic_visit to check it is not used.
generic_visit_args	Overrides generic_visit to keep track of the indentation and the node parent. The function will add field …
make_msg	Displays line and column information into a string.
print_tree	Displays the tree of instructions.
push	Pushes an element into a list.
visit	Visits a node, a method must exist for every object class.
visit_	If an element is not found…
visit_Add
visit_arg
visit_arguments
visit_Assign
visit_Attribute
visit_BinOp
visit_Call
visit_Compare
visit_Div
visit_FunctionDef
visit_Gt
visit_keyword
visit_List
visit_Load
visit_Lt
visit_MatMult
visit_Module
visit_Mult
visit_Name
visit_NameConstant	A name.
visit_Num
visit_Pow
visit_Return
visit_Store
visit_Str
visit_Sub
visit_UnaryOp
visit_USub

Documentation

One class which visits a syntax tree.

source on GitHub

class mlprodict.onnx_grammar.node_visitor_translator.CodeNodeVisitor(translator=None)

Bases: ast.NodeVisitor

Defines a visitor which walks though the syntax tree of the code.

Get the tree of a simple function

The following code uses Python syntax but follows a SQL logic.

<<<

import ast
import inspect
from textwrap import dedent
from mlprodict.onnx_grammar import CodeNodeVisitor


def norm2(x, y):
    delta = x - y
    n = delta ** 2
    return n


code = dedent(inspect.getsource(norm2))
node = ast.parse(code)
v = CodeNodeVisitor()
v.visit(node)
for r in v.Rows:
    print("{0}{1}: {2}".format("    " * r["indent"], r["type"], r["str"]))

>>>

    Module: 
        FunctionDef: norm2
            arguments: 
                arg: x
                arg: y
            Assign: 
                Name: delta
                BinOp: 
                    Name: x
                    Sub: 
                    Name: y
            Assign: 
                Name: n
                BinOp: 
                    Name: delta
                    Pow: 
                    Num: 2
            Return: 
                Name: n

source on GitHub

Parameters

translator – CodeTranslator

By default the translator is OnnxTranslator.

source on GitHub

property Rows

returns a list of dictionaries with all the elements of the code

source on GitHub

__init__(translator=None)

Parameters

translator – CodeTranslator

By default the translator is OnnxTranslator.

source on GitHub

export(context=None, **kwargs)

Calls method export from the translator class.

Parameters

• context – known python needed to run the translated function

• kwargs – whatever the method export from the translator class ingests

Returns

whatever the method export from the translator class returns

source on GitHub

generic_visit(node)

Overrides generic_visit to check it is not used.

source on GitHub

generic_visit_args(node, row)

Overrides generic_visit to keep track of the indentation and the node parent. The function will add field row["children"] = visited nodes from here.

Parameters

• node – node which needs to be visited

• row – row (a dictionary)

Returns

See ast.NodeVisitor.generic_visit

source on GitHub

make_msg(node)

Displays line and column information into a string.

source on GitHub

static print_node(node)

Debugging purpose.

source on GitHub

print_tree()

Displays the tree of instructions.

Returns

string

source on GitHub

push(row)

Pushes an element into a list.

source on GitHub

visit(node)

Visits a node, a method must exist for every object class.

source on GitHub

visit_(node)

If an element is not found…

source on GitHub

visit_NameConstant(node)

A name.

source on GitHub