Struct ResNetJit 

pub struct ResNetJit { /* private fields */ }

Implementations

impl ResNetJit

pub fn new(model: ResNet) -> Self

pub fn with_b_bound(self, max: usize) -> Self

Override the upper bound for the b symbolic variable. Must be called before prepare/prepare_with_config. Panics if max < min.

pub fn with_b_min_bound(self, min: usize) -> Self

Override the lower bound for the b symbolic variable. Must be called before prepare/prepare_with_config. Panics if min > max.

pub fn with_b_fixed(self, value: usize) -> Self

Pin b to a single value, making it a JIT-time constant. Sets both bounds to value so only value is accepted at execute time. Must be called before prepare/prepare_with_config. Panics on value == 0.

pub fn prepare(&mut self, images: InputSpec) -> Result<()>

pub fn prepare_with_config( &mut self, images: InputSpec, config: &PrepareConfig, ) -> Result<()>

pub fn images_mut(&mut self) -> Result<&mut Buffer>

pub fn output(&self) -> Result<&Buffer>

pub fn buffers(&self) -> Result<&[Buffer]>

pub fn output_buffers(&self) -> Result<Vec<&Buffer>>

pub fn input_buffer_ids(&self) -> Result<Vec<BufferId>>

pub fn prepared_kernels(&self) -> Result<Vec<&PreparedKernel>>

pub fn execute(&mut self) -> Result<()>

pub fn execute_profiled(&mut self) -> Result<Vec<KernelProfile>>

pub fn execute_with_vars(&mut self, vars: &[(&str, i64)]) -> Result<()>

pub fn execute_with_vars_profiled( &mut self, vars: &[(&str, i64)], ) -> Result<Vec<KernelProfile>>