Merge pull request #328 from dskkato/tensorflow-sys/expose_eager_api

[tensorflow-sys] Expose the eager API.

Cargo.toml

@@ -40,6 +40,7 @@ image = "0.23.14"
 [features]
 tensorflow_gpu = ["tensorflow-sys/tensorflow_gpu"]
 tensorflow_unstable = []
+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 = [] # DO NOT RELY ON THIS

tensorflow-sys/Cargo.toml

@@ -31,6 +31,7 @@ zip = "0.5.13"
 
 [features]
 tensorflow_gpu = []
+eager = []
 # This is for testing purposes; users should not use this.
 examples_system_alloc = []
 private-docs-rs = [] # DO NOT RELY ON THIS

tensorflow-sys/generate_bindgen_rs.sh

@@ -7,12 +7,14 @@ if ! which bindgen > /dev/null; then
     exit 1
 fi
 
-# See https://github.com/servo/rust-bindgen/issues/550 as to why
-# this is blacklisted.
-bindgen_options="--blacklist-type max_align_t --size_t-is-usize --default-enum-style=rust --generate-inline-functions"
-# TODO: revert
 include_dir="$HOME/git/tensorflow"
 
-cmd="bindgen ${bindgen_options} ${include_dir}/tensorflow/c/c_api.h --output src/bindgen.rs --  -I ${include_dir}"
+bindgen_options_c_api="--allowlist-function TF_.+ --allowlist-type TF_.+ --allowlist-var TF_.+ --size_t-is-usize --default-enum-style=rust --generate-inline-functions"
+cmd="bindgen ${bindgen_options_c_api} ${include_dir}/tensorflow/c/c_api.h --output src/c_api.rs --  -I ${include_dir}"
 echo ${cmd}
 ${cmd}
+
+bindgen_options_eager="--allowlist-function TFE_.+ --allowlist-type TFE_.+ --allowlist-var TFE_.+ --blocklist-type TF_.+ --size_t-is-usize --default-enum-style=rust --generate-inline-functions"
+cmd="bindgen ${bindgen_options_eager} ${include_dir}/tensorflow/c/eager/c_api.h --output src/eager/c_api.rs --  -I ${include_dir}"
+echo ${cmd}
+${cmd}