Struct BuilderConfig 

pub struct BuilderConfig(/* private fields */);

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 with_optimization_profile( self, optimization_profile: OptimizationProfile<'_>, ) -> Result<Self, Error>

Add an optimization profile.

TensorRT documentation

pub fn add_optimization_profile( &mut self, optimization_profile: OptimizationProfile<'_>, ) -> Result<(), Error>

Add an optimization profile.

TensorRT documentation

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.