From 17f3e8ad39919a32db99ff0666c1954d92490987 Mon Sep 17 00:00:00 2001 From: Juhyun Lee Date: Tue, 3 Dec 2019 13:37:51 -0800 Subject: [PATCH] Make a couple of targets in grappler buildable for Android. PiperOrigin-RevId: 283612146 Change-Id: If64befd6fcd53500413c98c0978111b5abc06bcd --- tensorflow/core/grappler/BUILD | 18 ++++++++++++------ tensorflow/core/grappler/utils/BUILD | 14 ++++++++++---- 2 files changed, 22 insertions(+), 10 deletions(-) diff --git a/tensorflow/core/grappler/BUILD b/tensorflow/core/grappler/BUILD index 3f79c023caf..fd2ea4f893c 100644 --- a/tensorflow/core/grappler/BUILD +++ b/tensorflow/core/grappler/BUILD @@ -23,14 +23,20 @@ cc_library( hdrs = ["utils.h"], visibility = ["//visibility:public"], deps = [ - "//tensorflow/core:framework", - "//tensorflow/core:graph", - "//tensorflow/core:lib", - "//tensorflow/core:lib_internal", - "//tensorflow/core:protos_all_cc", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", - ], + ] + select({ + "//tensorflow:android": [ + "//tensorflow/core:android_tensorflow_lib", + ], + "//conditions:default": [ + "//tensorflow/core:framework", + "//tensorflow/core:graph", + "//tensorflow/core:lib", + "//tensorflow/core:lib_internal", + "//tensorflow/core:protos_all_cc", + ], + }), ) tf_cc_test( diff --git a/tensorflow/core/grappler/utils/BUILD b/tensorflow/core/grappler/utils/BUILD index 8941d5552b6..7572141d415 100644 --- a/tensorflow/core/grappler/utils/BUILD +++ b/tensorflow/core/grappler/utils/BUILD @@ -386,11 +386,17 @@ cc_library( hdrs = ["transitive_fanin.h"], visibility = ["//visibility:public"], deps = [ - "//tensorflow/core:framework", - "//tensorflow/core:lib", - "//tensorflow/core:protos_all_cc", "//tensorflow/core/grappler:utils", - ], + ] + select({ + "//tensorflow:android": [ + "//tensorflow/core:android_tensorflow_lib", + ], + "//conditions:default": [ + "//tensorflow/core:framework", + "//tensorflow/core:lib", + "//tensorflow/core:protos_all_cc", + ], + }), ) tf_cc_test(