pub struct NeuralDecoder { /* private fields */ }Expand description
Neural decoder for model decompression
Implementations§
Source§impl NeuralDecoder
impl NeuralDecoder
Sourcepub fn new(config: DecoderConfig, codebooks: LayerCodebook) -> Self
pub fn new(config: DecoderConfig, codebooks: LayerCodebook) -> Self
Create a new decoder with codebooks
Sourcepub fn decode_tensor(&self, encoded: &EncodedTensor) -> Result<DecodedTensor>
pub fn decode_tensor(&self, encoded: &EncodedTensor) -> Result<DecodedTensor>
Decode an encoded tensor
Sourcepub fn decode_tensors(
&self,
encoded: &[EncodedTensor],
) -> Result<Vec<DecodedTensor>>
pub fn decode_tensors( &self, encoded: &[EncodedTensor], ) -> Result<Vec<DecodedTensor>>
Decode multiple tensors
Sourcepub fn decode_streaming<'a>(
&'a self,
encoded: &'a EncodedTensor,
chunk_size: usize,
) -> impl Iterator<Item = Result<Vec<f32>>> + 'a
pub fn decode_streaming<'a>( &'a self, encoded: &'a EncodedTensor, chunk_size: usize, ) -> impl Iterator<Item = Result<Vec<f32>>> + 'a
Decode with streaming (yields chunks)
Sourcepub fn config(&self) -> &DecoderConfig
pub fn config(&self) -> &DecoderConfig
Get decoder configuration
Sourcepub fn estimate_throughput(&self, stats: &DecodingStats) -> f32
pub fn estimate_throughput(&self, stats: &DecodingStats) -> f32
Estimate decode throughput in GB/s
Trait Implementations§
Auto Trait Implementations§
impl Freeze for NeuralDecoder
impl RefUnwindSafe for NeuralDecoder
impl Send for NeuralDecoder
impl Sync for NeuralDecoder
impl Unpin for NeuralDecoder
impl UnsafeUnpin for NeuralDecoder
impl UnwindSafe for NeuralDecoder
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more