-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
31 lines (27 loc) · 874 Bytes
/
Cargo.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
[package]
name = "onnx-rust"
authors = ["Andrea S.", "Alberto V."]
description = "Library and CLI tool written in Rust, made to parse models saved using the ONNX standard and perform inference with them."
version = "0.1.0"
edition = "2021"
[dependencies]
clap = { version = "4.3.22", features = ["derive"] }
env_logger = "0.10.0"
itertools = "0.11.0"
log = "0.4.20"
ndarray = { version = "0.15.6", features = ["matrixmultiply-threading"] }
ndarray_einsum_beta = "0.7.0"
prettytable = "0.10.0"
rayon = "1.7.0"
pyo3 = { version = "0.19.1", features = ["extension-module"], optional = true }
numpy = { version = "0.19", optional = true }
pyo3-log = { version = "0.8.3", optional = true }
[features]
python = [ "dep:pyo3", "dep:numpy", "dep:pyo3-log" ]
[lib]
name = "onnx_rust"
crate-type = ["cdylib","lib"]
path = "src/lib.rs"
[[bin]]
name = "onnx-cli"
path = "src/main.rs"