pub fn compile_model(
arch: Arch,
config_path: Option<&Path>,
weights_path: &Path,
batch: usize,
seq: usize,
device: Device,
) -> Result<(usize, CompiledGraph, HashMap<String, Vec<f32>>)>Expand description
Compile an embedding graph for the given batch/seq on device.