Skip to main content

compile_model_cpu

Function compile_model_cpu 

Source
pub fn compile_model_cpu(
    arch: Arch,
    config_path: Option<&Path>,
    weights_path: &Path,
    batch: usize,
    seq: usize,
) -> Result<(usize, CompiledGraph, HashMap<String, Vec<f32>>)>
Expand description

Compile on CPU (convenience for tests and default RlxEmbed::from_dir).