diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-03-27 11:11:34 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-03-27 12:26:57 -0700 |
commit | ca6b88eb95b089010a1b970e0de7398195b5bcca (patch) | |
tree | f1008833be270c9e05419f74644c9b859237ec97 /tensorflow/cc/saved_model | |
parent | f12b26ea5b6953a16c3a968aaa0fd623b1f7a74f (diff) |
Add support for SavedModel on Android.
Change: 151352935
Diffstat (limited to 'tensorflow/cc/saved_model')
-rw-r--r-- | tensorflow/cc/saved_model/BUILD | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/tensorflow/cc/saved_model/BUILD b/tensorflow/cc/saved_model/BUILD index f1595e43d7..1cc7cf3f20 100644 --- a/tensorflow/cc/saved_model/BUILD +++ b/tensorflow/cc/saved_model/BUILD @@ -9,7 +9,14 @@ licenses(["notice"]) # Apache 2.0 exports_files(["LICENSE"]) -load("//tensorflow:tensorflow.bzl", "tf_cc_test") +load( + "//tensorflow:tensorflow.bzl", + "if_android", + "if_ios", + "if_mobile", + "if_not_mobile", + "tf_cc_test", +) cc_library( name = "constants", @@ -31,11 +38,14 @@ cc_library( hdrs = ["loader.h"], deps = [ ":loader_lite", + ] + if_not_mobile([ "//tensorflow/core:core_cpu", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", "//tensorflow/core:tensorflow", - ], + ]) + if_android([ + "//tensorflow/core:android_tensorflow_lib", + ]), ) cc_library( @@ -44,12 +54,14 @@ cc_library( hdrs = ["loader.h"], deps = [ ":constants", + ] + if_not_mobile([ "//tensorflow/core:core_cpu", "//tensorflow/core:lib", "//tensorflow/core:lib_internal", "//tensorflow/core:protos_all_cc", "//tensorflow/core/util/tensor_bundle:naming", - ], + # mobile not supported yet + ]), ) tf_cc_test( |