Struct ROCm 

pub struct ROCm { /* private fields */ }

Available on crate feature rocm only.

Implementations

impl ROCm

pub fn build(self) -> ExecutionProviderDispatch

impl ROCm

pub fn with_device_id(self, device_id: i32) -> Self

pub fn with_exhaustive_conv_search(self, enable: bool) -> Self

pub fn with_conv_use_max_workspace(self, enable: bool) -> Self

pub fn with_mem_limit(self, limit: usize) -> Self

pub fn with_arena_extend_strategy(self, strategy: ArenaExtendStrategy) -> Self

pub fn with_copy_in_default_stream(self, enable: bool) -> Self

pub fn with_compute_stream(self, ptr: *mut c_void) -> Self

pub fn with_hip_graph(self, enable: bool) -> Self

pub fn with_tunable_op(self, enable: bool) -> Self

pub fn with_tuning(self, enable: bool) -> Self

pub fn with_max_tuning_duration(self, ms: i32) -> Self

Trait Implementations

impl ArbitrarilyConfigurableExecutionProvider for ROCm

fn with_arbitrary_config(self, key: impl ToString, value: impl ToString) -> Self

impl Clone for ROCm

fn clone(&self) -> ROCm

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ROCm

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for ROCm

fn default() -> ROCm

Returns the “default value” for a type.

impl ExecutionProvider for ROCm

fn name(&self) -> &'static str

Returns the identifier of this execution provider used internally by ONNX Runtime.

fn register(&self, session_builder: &mut SessionBuilder) -> Result<()>

Attempts to register this execution provider on the given session.

fn is_available(&self) -> Result<bool>

Returns Ok(true) if ONNX Runtime was compiled with support for this execution provider, and Ok(false) otherwise.

impl From<ROCm> for ExecutionProviderDispatch

fn from(value: ROCm) -> Self

Converts to this type from the input type.