diff options
author | Jonathan Hseu <jhseu@google.com> | 2016-12-19 13:02:20 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-19 13:21:59 -0800 |
commit | 35ea93ba205d44319f408a7d90dc385fd86d6b56 (patch) | |
tree | 2e4f0a5c0ab5765c78a246c99654946ee3a884d8 /tensorflow/c/BUILD | |
parent | a325547f4c7bdb169c62bdffa894617beffa7e5f (diff) |
Add SavedModel support to the C API
Change: 142477558
Diffstat (limited to 'tensorflow/c/BUILD')
-rw-r--r-- | tensorflow/c/BUILD | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/tensorflow/c/BUILD b/tensorflow/c/BUILD index 7afac44452..32696e83e4 100644 --- a/tensorflow/c/BUILD +++ b/tensorflow/c/BUILD @@ -25,6 +25,7 @@ tf_cuda_library( hdrs = ["c_api.h"], visibility = ["//visibility:public"], deps = [ + "//tensorflow/cc/saved_model:loader", "//tensorflow/core:core_cpu", "//tensorflow/core:framework", "//tensorflow/core:lib", @@ -62,7 +63,10 @@ tf_cc_test( name = "c_api_test", size = "small", srcs = ["c_api_test.cc"], - data = [":test_op.so"], + data = [ + ":test_op.so", + "//tensorflow/cc/saved_model:saved_model_half_plus_two", + ], linkopts = select({ "//tensorflow:darwin": ["-headerpad_max_install_names"], "//conditions:default": [], @@ -70,6 +74,9 @@ tf_cc_test( linkstatic = tf_kernel_tests_linkstatic(), deps = [ ":c_api", + "//tensorflow/cc/saved_model:signature_constants", + "//tensorflow/cc/saved_model:tag_constants", + "//tensorflow/core:core_cpu_internal", "//tensorflow/core:direct_session", "//tensorflow/core:framework", "//tensorflow/core:lib", |