ai.onnxruntime.providers

Class OrtTensorRTProviderOptions

java.lang.Object

ai.onnxruntime.OrtProviderOptions

ai.onnxruntime.providers.OrtTensorRTProviderOptions

All Implemented Interfaces:

AutoCloseable

public final class OrtTensorRTProviderOptions
extends OrtProviderOptions

Options for configuring the TensorRT execution provider.

Supported options are listed on the ORT website.

Nested Class Summary

Nested classes/interfaces inherited from class ai.onnxruntime.OrtProviderOptions

OrtProviderOptions.OrtProviderSupplier

Field Summary

Fields

Modifier and Type	Field and Description
protected Map<String,String>	options A Java side copy of the options.

Fields inherited from class ai.onnxruntime.OrtProviderOptions

closed, nativeHandle

Constructor Summary

Constructors

Constructor and Description
OrtTensorRTProviderOptions() Constructs TensorRT execution provider options for device 0.
OrtTensorRTProviderOptions(int deviceId) Constructs TensorRT execution provider options for the specified non-negative device id.

Method Summary

Modifier and Type	Method and Description
void	add(String key, String value) Adds a configuration option to this options.
protected void	applyToNative() Applies the Java side configuration to the native side object.
protected void	applyToNative(long apiHandle, long nativeHandle, String[] keys, String[] values) Adds the options to this options instance.
protected void	close(long apiHandle, long nativeHandle) Closes this options instance.
String	getOptionsString() Returns the serialized options string
OrtProvider	getProvider() Gets the provider enum for this options instance.
void	parseOptionsString(String serializedForm) Parses the output of getOptionsString() and adds those options to this options instance.
String	toString()

Methods inherited from class ai.onnxruntime.OrtProviderOptions

checkClosed, close, getApiHandle, isClosed, loadLibraryAndCreate

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Field Detail

options

protected final Map<String,String> options

A Java side copy of the options.

Constructor Detail

OrtTensorRTProviderOptions

public OrtTensorRTProviderOptions()
                           throws OrtException

Constructs TensorRT execution provider options for device 0.

Throws:

OrtException - If TensorRT is unavailable.

OrtTensorRTProviderOptions

public OrtTensorRTProviderOptions(int deviceId)
                           throws OrtException

Constructs TensorRT execution provider options for the specified non-negative device id.

Parameters:

deviceId - The device id.

Throws:

OrtException - If TensorRT is unavailable.

Method Detail

getProvider

public OrtProvider getProvider()

Description copied from class: OrtProviderOptions

Gets the provider enum for this options instance.

Specified by:

getProvider in class OrtProviderOptions

Returns:

The provider enum.

applyToNative

protected void applyToNative(long apiHandle,
                             long nativeHandle,
                             String[] keys,
                             String[] values)
                      throws OrtException

Adds the options to this options instance.

Parameters:

apiHandle - The api pointer.

nativeHandle - The native options pointer.

keys - The option keys.

values - The option values.

Throws:

OrtException - If the addition failed.

close

protected void close(long apiHandle,
                     long nativeHandle)

Closes this options instance.

Specified by:

close in class OrtProviderOptions

Parameters:

apiHandle - The api pointer.

nativeHandle - The native options pointer.

add

public void add(String key,
                String value)
         throws OrtException

Adds a configuration option to this options.

Parameters:

key - The key.

value - The value.

Throws:

OrtException - If the addition failed.

parseOptionsString

public void parseOptionsString(String serializedForm)
                        throws OrtException

Parses the output of getOptionsString() and adds those options to this options instance.

Parameters:

serializedForm - The serialized form to parse.

Throws:

OrtException - If the option could not be added.

toString

public String toString()

Overrides:

toString in class Object

getOptionsString

public String getOptionsString()

Returns the serialized options string

Returns:

The serialized options string.

applyToNative

protected void applyToNative()
                      throws OrtException

Description copied from class: OrtProviderOptions

Applies the Java side configuration to the native side object.

Specified by:

applyToNative in class OrtProviderOptions

Throws:

OrtException - If the native call failed.