onnxruntime helpers

Frequent functions

onnxruntime.get_device() → str

Return the device used to compute the prediction (CPU, MKL, …)

onnxruntime.capi._pybind_state.set_seed(arg0: int) → None

Sets the seed used for random number generation in Onnxruntime.

onnxruntime.capi._pybind_state.set_default_logger_severity(arg0: int) → None

Sets the default logging severity. 0:Verbose, 1:Info, 2:Warning, 3:Error, 4:Fatal

onnxruntime.capi._pybind_state.get_all_providers() → List[str]

Return list of Execution Providers that this version of Onnxruntime can support. The order of elements represents the default priority order of Execution Providers from highest to lowest.

Python Wrapper OrtDevice

class onnxruntime.capi._pybind_state.OrtDevice(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice, arg0: int, arg1: int, arg2: int) → None

ONNXRuntime device informaion.

__init__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice, arg0: int, arg1: int, arg2: int) → None
static cpu() → int
static cuda() → int
static default_memory() → int
device_id(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice) → int

Device Id.

device_type(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice) → int

Device Type.

C class, OrtDevice

class onnxruntime.capi._pybind_state.OrtDevice(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice, arg0: int, arg1: int, arg2: int) → None

ONNXRuntime device informaion.

__init__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice, arg0: int, arg1: int, arg2: int) → None
static cpu() → int
static cuda() → int
static default_memory() → int
device_id(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice) → int

Device Id.

device_type(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtDevice) → int

Device Type.

C classes, frequent types

class onnxruntime.capi._pybind_state.ModelMetadata

Pre-defined and custom metadata about the model. It is usually used to identify the model used to run the prediction and facilitate the comparison.

__init__(*args, **kwargs)
property custom_metadata_map

additional metadata

property description

description of the model

property domain

ONNX domain

property graph_description

description of the graph hosted in the model

property graph_name

graph name

property producer_name

producer name

property version

version of the model

class onnxruntime.capi._pybind_state.OrtMemType(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtMemType, value: int) → None

Members:

CPU_INPUT

CPU_OUTPUT

CPU

DEFAULT

CPU = <OrtMemType.CPU_OUTPUT: -1>
CPU_INPUT = <OrtMemType.CPU_INPUT: -2>
CPU_OUTPUT = <OrtMemType.CPU_OUTPUT: -1>
DEFAULT = <OrtMemType.DEFAULT: 0>
__eq__(self: object, other: object) → bool
__getstate__(self: object) → int
__hash__(self: object) → int
__index__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtMemType) → int
__init__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtMemType, value: int) → None
__int__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtMemType) → int
__members__ = {'CPU': <OrtMemType.CPU_OUTPUT: -1>, 'CPU_INPUT': <OrtMemType.CPU_INPUT: -2>, 'CPU_OUTPUT': <OrtMemType.CPU_OUTPUT: -1>, 'DEFAULT': <OrtMemType.DEFAULT: 0>}
__ne__(self: object, other: object) → bool
__repr__(self: object) → str
__setstate__(self: onnxruntime.capi.onnxruntime_pybind11_state.OrtMemType, state: int) → None
__str__()

name(self: handle) -> str

property name
property value

Rare functions

onnxruntime.capi._pybind_state.enable_telemetry_events() → None

Enables platform-specific telemetry collection where applicable.

onnxruntime.capi._pybind_state.disable_telemetry_events() → None

Disables platform-specific telemetry collection.

onnxruntime.capi._pybind_state.create_and_register_allocator(arg0: OrtMemoryInfo, arg1: OrtArenaCfg) → None

previous

Selected onnxruntime-training API

next

OrtValue