AST#
Main functions#
mlprodict.onnx_tools.onnx_grammar.translate_fct2onnx (fct, context = None, cpl = False, context_cpl = None, output_names = None, dtype = <class ‘numpy.float32’>, verbose = 0, fLOG = None)
Translates a function into ONNX. The code it produces is using classes OnnxAbs, OnnxAdd, …
Additional functions#
mlprodict.onnx_tools.onnx_grammar.onnx_translation.get_default_context ()
mlprodict.onnx_tools.onnx_grammar.onnx_translation.get_default_context_cpl ()
Returns a default useful context to compile the converter returned by
translate_fct2onnx.
mlprodict.onnx_tools.onnx_grammar.onnx_translation.py_make_float_array (cst, op_version = None)
Creates an array with a single element from a constant.
mlprodict.onnx_tools.onnx_grammar.onnx_translation.py_opp (x, op_version = None)
Function for python unary operator
-.
mlprodict.onnx_tools.onnx_grammar.onnx_translation.py_mul (x, op_version = None)
Function for python operator
*.
mlprodict.onnx_tools.onnx_grammar.onnx_translation.py_pow (x, p, op_version = None)
Function for python operator
**.
mlprodict.onnx_tools.onnx_grammar.onnx_translation.squareform_pdist (X, metric = ‘sqeuclidean’, op_version = None)
Replacements for squareform and pdist.
Grammar Objects#
mlprodict.onnx_tools.onnx_grammar.CodeNodeVisitor (self, translator = None)
Defines a visitor which walks though the syntax tree of the code.
mlprodict.onnx_tools.onnx_grammar.onnx_translator.CodeTranslator (self, visitor)
Class which converts a Python function into something else. It must implements methods visit and depart.
mlprodict.onnx_tools.onnx_grammar.onnx_translator.OnnxTranslator (self, visitor)
Class which converts a Python function into an ONNX function. It must implements methods visit and depart.