candle_onnx/lib.rs
1use candle::Result;
2use prost::Message;
3
4pub mod onnx {
5 include!(concat!(env!("OUT_DIR"), "/onnx.rs"));
6}
7
8pub mod eval;
9pub use eval::{dtype, simple_eval};
10
11pub fn read_file<P: AsRef<std::path::Path>>(p: P) -> Result<onnx::ModelProto> {
12 let buf = std::fs::read(p)?;
13 onnx::ModelProto::decode(buf.as_slice()).map_err(candle::Error::wrap)
14}