mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 19:47:12 +00:00
Improve the ONNX basic example + bugfixes (#1266)
* Generate some zeros tensor in the onnx simple-eval example. * Fix the casting operation. * Support more ops. * Handle reshape. * Concat. * Softmax.
This commit is contained in:
@ -6,7 +6,7 @@ pub mod onnx {
|
||||
}
|
||||
|
||||
mod eval;
|
||||
pub use eval::simple_eval;
|
||||
pub use eval::{dtype, simple_eval};
|
||||
|
||||
pub fn read_file<P: AsRef<std::path::Path>>(p: P) -> Result<onnx::ModelProto> {
|
||||
let buf = std::fs::read(p)?;
|
||||
|
Reference in New Issue
Block a user