forked from tensorflow/rust
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Cargo.toml
72 lines (59 loc) · 1.95 KB
/
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
[package]
name = "tensorflow"
version = "0.19.1"
authors = ["Adam Crume <acrume@google.com>"]
description = "Rust language bindings for TensorFlow."
license = "Apache-2.0"
keywords = ["TensorFlow", "bindings"]
readme = "README.md"
repository = "https://github.com/tensorflow/rust"
documentation = "https://tensorflow.github.io/rust/tensorflow"
edition = "2018"
rust-version = "1.59.0"
[lib]
# We want to run doctests, but they're broken by https://github.com/rust-lang/cargo/issues/8531.
doctest = false
# Prevent downloading or building TensorFlow when building docs on docs.rs.
[package.metadata.docs.rs]
features = ["private-docs-rs", "tensorflow_unstable", "ndarray", "eager"]
[dependencies]
libc = "0.2.132"
num-complex = { version = "0.4.2", default-features = false }
tensorflow-internal-macros = { version = "=0.0.3", path = "tensorflow-internal-macros" }
tensorflow-sys = { version = "0.22.1", path = "tensorflow-sys" }
byteorder = "1.4.3"
crc = "3.0.0"
half = "2.1.0"
# This is used internally but not intended to be exposed through the API.
protobuf = "=2.27.1"
# Enables conversions between ndarray::Array objects and tensorflow::Tensor
ndarray = { version = "0.15.6", optional = true }
rustversion = "1.0.9"
[dev-dependencies]
random = "0.12.2"
serial_test = "0.9.0"
[features]
tensorflow_gpu = ["tensorflow-sys/tensorflow_gpu"]
tensorflow_unstable = []
tensorflow_runtime_linking = ["tensorflow-sys/runtime_linking"]
eager = ["tensorflow-sys/eager"]
# This is for testing purposes; users should not use this.
examples_system_alloc = ["tensorflow-sys/examples_system_alloc"]
private-docs-rs = ["tensorflow-sys/private-docs-rs"] # DO NOT RELY ON THIS
[workspace]
[[example]]
name = "addition"
[[example]]
name = "expressions"
required-features = ["tensorflow_unstable"]
[[example]]
name = "regression"
[[example]]
name = "regression_savedmodel"
[[example]]
name = "regression_checkpoint"
[[example]]
name = "xor"
[[example]]
name = "mobilenetv3"
required-features = ["eager"]