Struct CpuBurnDecoder 

pub struct CpuBurnDecoder<B: Backend> { /* private fields */ }

Persistent CPU codec session followed by one Burn materialization per output group.

Implementations

impl<B: Backend> CpuBurnDecoder<B>

pub fn new(device: B::Device, options: BatchDecodeOptions) -> Self

Construct a CPU codec session for one Burn device.

pub const fn codec(&self) -> &CpuBatchDecoder

Borrow the retained codec session.

pub const fn device(&self) -> &B::Device

Borrow the Burn device receiving decoded tensors.

pub fn prepare( &self, inputs: Vec<EncodedImage>, ) -> Result<PreparedBatch, BurnDecodeError>

Parse and group owned inputs once for repeated training epochs.

pub fn prepare_prepared_images( &self, images: Vec<PreparedImage>, ) -> Result<PreparedBatch, BurnDecodeError>

Regroup caller-supplied prepared images without reparsing encoded bytes.

Source indices in the returned batch are positions in images; each PreparedImage::source_index remains its original preparation index.

pub fn decode( &mut self, inputs: Vec<EncodedImage>, ) -> Result<BurnBatchDecode<B>, BurnDecodeError>

Prepare, decode, and materialize one owned batch.

pub fn decode_prepared_images( &mut self, images: Vec<PreparedImage>, ) -> Result<BurnBatchDecode<B>, BurnDecodeError>

Regroup, decode, and materialize caller-supplied prepared images.

pub fn decode_prepared( &mut self, prepared: &PreparedBatch, ) -> Result<BurnBatchDecode<B>, BurnDecodeError>

Decode a reusable prepared batch into ordinary rank-4 Burn integer tensors.

Trait Implementations

impl<B: Debug + Backend> Debug for CpuBurnDecoder<B>

where B::Device: Debug,

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

Formats the value using the given formatter.