diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-19 08:52:49 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-19 08:58:53 -0700 |
commit | e9e48b963b1ad1274ad8a0ad7d07d7fa990fe6b9 (patch) | |
tree | 26a7a7c45908a462c4d317cdf39f8b6349459d67 | |
parent | daafc6571a4817b1313b7c243fbd35e3a9f12dab (diff) |
Update `reader` dependencies such that the SavedModel loader still works on mobile.
PiperOrigin-RevId: 205248073
-rw-r--r-- | tensorflow/cc/saved_model/BUILD | 13 |
1 files changed, 12 insertions, 1 deletions
diff --git a/tensorflow/cc/saved_model/BUILD b/tensorflow/cc/saved_model/BUILD index 730b1b669b..3d3895c8fa 100644 --- a/tensorflow/cc/saved_model/BUILD +++ b/tensorflow/cc/saved_model/BUILD @@ -39,9 +39,20 @@ cc_library( hdrs = ["reader.h"], deps = [ ":constants", + ] + if_not_mobile([ + # TODO(b/111634734): :lib and :protos_all contain dependencies that + # cannot be built on mobile platforms. Instead, include the appropriate + # tf_lib depending on the build platform. "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", - ], + ]) + if_mobile([ + # Mobile-friendly SavedModel proto. See go/portable-proto for more info. + "//tensorflow/core:saved_model_portable_proto", + ]) + if_android([ + "//tensorflow/core:android_tensorflow_lib", + ]) + if_ios([ + "//tensorflow/core:ios_tensorflow_lib", + ]), ) tf_cc_test( |