Struct async_tensorrt::ffi::builder_config::BuilderConfig

pub struct BuilderConfig(_);

Holds properties for configuring a builder to produce an engine.

TensorRT documentation

Implementations

impl BuilderConfig

pub fn with_max_workspace_size(self, size: usize) -> Self

Set the maximum workspace size.

TensorRT documentation

Arguments

• size - The maximum GPU temporary memory which the engine can use at execution time in bytes.

pub fn with_strict_types(self) -> Self

Set the kSTRICT_TYPES flag.

TensorRT documentation for setFlag TensorRT documentation for kSTRICT_TYPES

pub fn with_fp16(self) -> Self

Set the kFP16 flag.

TensorRT documentation for setFlag TensorRT documentation for kFP16

pub fn as_ptr(&self) -> *const c_void

Get internal readonly pointer.

pub fn as_mut_ptr(&mut self) -> *mut c_void

Get internal mutable pointer.

Trait Implementations

impl Drop for BuilderConfig

fn drop(&mut self)

Executes the destructor for this type.

impl Send for BuilderConfig

Implements Send for BuilderConfig.

Safety

The TensorRT API is thread-safe with regards to all operations on BuilderConfig.

impl Sync for BuilderConfig

Implements Sync for BuilderConfig.

Safety

The TensorRT API is thread-safe with regards to all operations on BuilderConfig.