提交 51c0ce54 编写于 作者: A A. Unique TensorFlower 提交者: TensorFlower Gardener

Pull some common lists of headers and deps out of the common android libraries.

Change: 118427688
上级 f4acbbf2
...@@ -537,6 +537,8 @@ tf_cuda_library( ...@@ -537,6 +537,8 @@ tf_cuda_library(
# Android-specific BUILD targets # Android-specific BUILD targets
load("//tensorflow:tensorflow.bzl", "tf_android_core_proto_sources") load("//tensorflow:tensorflow.bzl", "tf_android_core_proto_sources")
load("//tensorflow:tensorflow.bzl", "tf_android_core_proto_sources_relative")
load("//tensorflow:tensorflow.bzl", "tf_android_core_proto_headers")
# List of protos we want on android # List of protos we want on android
filegroup( filegroup(
...@@ -620,6 +622,16 @@ ANDROID_TF_COPTS = [ ...@@ -620,6 +622,16 @@ ANDROID_TF_COPTS = [
"-DTF_LEAN_BINARY", "-DTF_LEAN_BINARY",
] ]
ANDROID_TF_LITE_HDRS = [
"framework/tensor.h",
"platform/default/integral_types.h",
"platform/env.h",
"platform/logging.h",
"platform/platform.h",
"platform/types.h",
"public/session.h",
]
# Native library support for Android applications. # Native library support for Android applications.
# Does not contain operators, use :android_tensorflow_lib if you want full # Does not contain operators, use :android_tensorflow_lib if you want full
# operator support. # operator support.
...@@ -633,19 +645,12 @@ ANDROID_TF_COPTS = [ ...@@ -633,19 +645,12 @@ ANDROID_TF_COPTS = [
# --host_crosstool_top=@bazel_tools//tools/cpp:toolchain # --host_crosstool_top=@bazel_tools//tools/cpp:toolchain
cc_library( cc_library(
name = "android_tensorflow_lib_lite", name = "android_tensorflow_lib_lite",
srcs = select({ srcs =
":android": ["//tensorflow/core:android_srcs"], select({
"//conditions:default": [], ":android": ["//tensorflow/core:android_srcs"],
}), "//conditions:default": [],
hdrs = [ }),
"framework/tensor.h", hdrs = ANDROID_TF_LITE_HDRS,
"platform/default/integral_types.h",
"platform/env.h",
"platform/logging.h",
"platform/platform.h",
"platform/types.h",
"public/session.h",
],
copts = select({ copts = select({
":android": ANDROID_TF_COPTS + ["-Os"], ":android": ANDROID_TF_COPTS + ["-Os"],
"//conditions:default": [], "//conditions:default": [],
......
...@@ -8,31 +8,43 @@ load("//tensorflow/core:platform/default/build_config_root.bzl", ...@@ -8,31 +8,43 @@ load("//tensorflow/core:platform/default/build_config_root.bzl",
# List of proto files for android builds # List of proto files for android builds
def tf_android_core_proto_sources(): def tf_android_core_proto_sources():
return ["//tensorflow/core:" + p
for p in tf_android_core_proto_sources_relative()]
# As tf_android_core_proto_sources, but paths relative to
# //third_party/tensorflow/core.
def tf_android_core_proto_sources_relative():
return [ return [
"//tensorflow/core:example/example.proto", "example/example.proto",
"//tensorflow/core:example/feature.proto", "example/feature.proto",
"//tensorflow/core:framework/allocation_description.proto", "framework/allocation_description.proto",
"//tensorflow/core:framework/attr_value.proto", "framework/attr_value.proto",
"//tensorflow/core:framework/device_attributes.proto", "framework/device_attributes.proto",
"//tensorflow/core:framework/function.proto", "framework/function.proto",
"//tensorflow/core:framework/graph.proto", "framework/graph.proto",
"//tensorflow/core:framework/kernel_def.proto", "framework/kernel_def.proto",
"//tensorflow/core:framework/log_memory.proto", "framework/log_memory.proto",
"//tensorflow/core:framework/op_def.proto", "framework/op_def.proto",
"//tensorflow/core:framework/step_stats.proto", "framework/step_stats.proto",
"//tensorflow/core:framework/summary.proto", "framework/summary.proto",
"//tensorflow/core:framework/tensor.proto", "framework/tensor.proto",
"//tensorflow/core:framework/tensor_description.proto", "framework/tensor_description.proto",
"//tensorflow/core:framework/tensor_shape.proto", "framework/tensor_shape.proto",
"//tensorflow/core:framework/tensor_slice.proto", "framework/tensor_slice.proto",
"//tensorflow/core:framework/types.proto", "framework/types.proto",
"//tensorflow/core:framework/versions.proto", "framework/versions.proto",
"//tensorflow/core:lib/core/error_codes.proto", "lib/core/error_codes.proto",
"//tensorflow/core:protobuf/config.proto", "protobuf/config.proto",
"//tensorflow/core:protobuf/saver.proto", "protobuf/saver.proto",
"//tensorflow/core:util/saved_tensor_slice.proto", "util/saved_tensor_slice.proto",
] ]
# Returns the list of pb.h headers that are generated for
# tf_android_core_proto_sources().
def tf_android_core_proto_headers():
return ["//tensorflow/core/" + p.replace(".proto", ".pb.h")
for p in tf_android_core_proto_sources_relative()]
def if_cuda(a, b=[]): def if_cuda(a, b=[]):
return select({ return select({
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册