Struct ResNetConfig 

pub struct ResNetConfig {
    pub depth: ResNetDepth,
    pub output: OutputMode,
    pub max_batch_size: usize,
}

Fields

depth: ResNetDepth

output: OutputMode

max_batch_size: usize

Upper bound on the symbolic b variable exposed by the JIT wrapper. The prepared plan’s image buffer is allocated to max_batch_size; the per-call execute_with_vars(&[("b", actual)]) shrinks the batch dim to the live size.

Implementations

impl ResNetConfig

pub fn new(depth: ResNetDepth, output: OutputMode) -> Self

pub fn with_max_batch_size(self, max_batch_size: usize) -> Self

Trait Implementations

impl Clone for ResNetConfig

fn clone(&self) -> ResNetConfig

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for ResNetConfig

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

Formats the value using the given formatter.