module tools.onnx_inference_ort_helper

Short summary

module mlprodict.tools.onnx_inference_ort_helper

Helpers for onnxruntime.

source on GitHub

Functions

function	truncated documentation
device_to_providers	Returns the corresponding providers for a specific device.
get_ort_device	Converts device into C_OrtDevice.

Documentation

Helpers for onnxruntime.

source on GitHub

mlprodict.tools.onnx_inference_ort_helper.device_to_providers(device)

Returns the corresponding providers for a specific device.

Parameters

device – C_OrtDevice

Returns

providers

source on GitHub

mlprodict.tools.onnx_inference_ort_helper.get_ort_device(device)

Converts device into C_OrtDevice.

Parameters

device – any type

Returns

C_OrtDevice

Example:

get_ort_device('cpu')
get_ort_device('gpu')
get_ort_device('cuda')
get_ort_device('cuda:0')

source on GitHub

previous

module tools.model_info

next

module tools.ort_wrapper