Source code for mlprodict.onnxrt.ops_onnxruntime

# -*- encoding: utf-8 -*-
"""
Shortcut to *ops_cpu*.


:githublink:`%|py|6`
"""
from ._op import OpRunOnnxRuntime



[docs]def load_op(onnx_node, desc=None, options=None, variables=None, dtype=None):
    """
    Gets the operator related to the *onnx* node.

    :param      onnx_node:       :epkg:`onnx` node
    :param      desc:            internal representation
    :param      options:         runtime options
    :param      variables:       registered variables created by previous operators
    :param      dtype:           float computation type
    :return:                     runtime class


    :githublink:`%|py|19`
    """
    if desc is None:
        raise ValueError(  # pragma: no cover
            "desc should not be None.")
    return OpRunOnnxRuntime(onnx_node, desc, variables=variables,
                            dtype=dtype, **options)