Trait SaveHandler

Source
pub trait SaveHandler<E: Error> {
    // Required methods
    fn write_hyperparameters(&mut self, writer: &mut dyn Write) -> Result<(), E>;
    fn tensor_data(&mut self, tensor_name: &str) -> Result<TensorSaveInfo, E>;
}
Expand description

A handler for saving a GGML model.

Required Methods§

Source

fn write_hyperparameters(&mut self, writer: &mut dyn Write) -> Result<(), E>

Called when the hyperparameters must be written.

Source

fn tensor_data(&mut self, tensor_name: &str) -> Result<TensorSaveInfo, E>

Called when information for a tensor is to be written.

Implementors§