Skip to content

Commit

Permalink
re-enable build-time protobuf generator
Browse files Browse the repository at this point in the history
  • Loading branch information
kali committed Sep 30, 2019
1 parent 04d4ee4 commit 3a438e3
Show file tree
Hide file tree
Showing 10 changed files with 23 additions and 5,165 deletions.
4 changes: 2 additions & 2 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -152,11 +152,11 @@ easily as practical to small-ish devices (think 20$ boards).

# License

Note: files in the `protos/tensorflow` directory are copied from the
Note: files in the `tensorflow/protos` directory are copied from the
[TensorFlow](https://github.com/tensorflow/tensorflow) project and are not
covered by the following licence statement.

Note: files in the `protos/onnx` directory are copied from the
Note: files in the `onnx/protos` directory are copied from the
[ONNX](https://github.com/onnx/onnx) project and are not
covered by the following licence statement.

Expand Down
4 changes: 4 additions & 0 deletions onnx/Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -24,3 +24,7 @@ protobuf = "^2.9"
smallvec = "0.6"
tract-core = { path = "../core" }
tract-linalg = { path = "../linalg" }

[build-dependencies]
protoc-rust = "^2.9"

File renamed without changes.
File renamed without changes.
File renamed without changes.
18 changes: 17 additions & 1 deletion onnx/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,23 @@ extern crate tract_linalg;

pub mod model;
pub mod ops;
pub mod pb;
pub mod pb {
#![allow(unknown_lints)]

#![cfg_attr(rustfmt, rustfmt_skip)]

#![allow(box_pointers)]
#![allow(dead_code)]
#![allow(missing_docs)]
#![allow(non_camel_case_types)]
#![allow(non_snake_case)]
#![allow(non_upper_case_globals)]
#![allow(trivial_casts)]
#![allow(unsafe_code)]
#![allow(unused_imports)]
#![allow(unused_results)]
include!(concat!(env!("OUT_DIR"), "/protobuf-generated/onnx.rs"));
}
pub mod pb_helpers;
pub mod tensor;

Expand Down
Loading

0 comments on commit 3a438e3

Please sign in to comment.