169 lines
4.5 KiB
Python
169 lines
4.5 KiB
Python
load(
|
|
"@io_bazel_rules_closure//closure:defs.bzl",
|
|
"closure_proto_library",
|
|
)
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_cc_test",
|
|
"tf_copts",
|
|
)
|
|
load(
|
|
"//tensorflow/core/platform:build_config.bzl",
|
|
"tf_proto_library",
|
|
"tf_pyclif_proto_library",
|
|
)
|
|
|
|
package(
|
|
default_visibility = [
|
|
"//tensorflow/core:__subpackages__",
|
|
],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
# TODO(bmzhao): Refactor this target to use granular dependencies
|
|
# after stage 4 of the TF build refactor is complete:
|
|
# https://github.com/tensorflow/community/pull/179
|
|
cc_library(
|
|
name = "example_parser_configuration",
|
|
srcs = ["example_parser_configuration.cc"],
|
|
hdrs = ["example_parser_configuration.h"],
|
|
copts = tf_copts(),
|
|
linkstatic = 1,
|
|
visibility = ["//visibility:public"],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "feature_util",
|
|
srcs = ["feature_util.cc"],
|
|
hdrs = ["feature_util.h"],
|
|
deps = [
|
|
":example_protos_cc",
|
|
"//tensorflow/core/platform:protobuf",
|
|
"//tensorflow/core/platform:stringpiece",
|
|
"//tensorflow/core/platform:types",
|
|
"@com_google_absl//absl/base:core_headers",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
# TODO(bmzhao): Refactor this target to use granular dependencies
|
|
# after stage 4 of the TF build refactor is complete:
|
|
# https://github.com/tensorflow/community/pull/179
|
|
tf_cc_test(
|
|
name = "example_parser_configuration_test",
|
|
size = "small",
|
|
srcs = ["example_parser_configuration_test.cc"],
|
|
data = [":example_parser_configuration_testdata"],
|
|
deps = [
|
|
"//tensorflow/cc:cc_ops",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:direct_session_internal",
|
|
"//tensorflow/core:example_parser_configuration",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:ops",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
"//tensorflow/core/kernels:example_parsing_ops",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "example_parser_configuration_testdata",
|
|
srcs = [
|
|
"testdata/parse_example_graph_def.pbtxt",
|
|
],
|
|
)
|
|
|
|
# TODO(bmzhao): Split this target into separate tf_proto_libraries.
|
|
# This target is a holdover from the target tensorflow/core:example_protos,
|
|
# but splitting it while require multiple LSCs.
|
|
tf_proto_library(
|
|
name = "example_protos",
|
|
srcs = [
|
|
"example.proto",
|
|
"feature.proto",
|
|
],
|
|
cc_api_version = 2,
|
|
make_default_target_header_only = True,
|
|
)
|
|
|
|
tf_proto_library(
|
|
name = "example_parser_configuration_proto",
|
|
srcs = ["example_parser_configuration.proto"],
|
|
cc_api_version = 2,
|
|
make_default_target_header_only = True,
|
|
protodeps = [
|
|
"//tensorflow/core/framework:tensor_shape_proto",
|
|
"//tensorflow/core/framework:tensor_proto",
|
|
"//tensorflow/core/framework:types_proto",
|
|
],
|
|
)
|
|
|
|
tf_proto_library(
|
|
name = "protos_all",
|
|
cc_api_version = 2,
|
|
make_default_target_header_only = True,
|
|
protodeps = [
|
|
":example_protos",
|
|
":example_parser_configuration_proto",
|
|
],
|
|
)
|
|
|
|
tf_pyclif_proto_library(
|
|
name = "example_pyclif",
|
|
proto_lib = "//tensorflow/core:protos_all",
|
|
proto_srcfile = "example.proto",
|
|
)
|
|
|
|
tf_pyclif_proto_library(
|
|
name = "feature_pyclif",
|
|
proto_lib = "//tensorflow/core:protos_all",
|
|
proto_srcfile = "feature.proto",
|
|
)
|
|
|
|
closure_proto_library(
|
|
name = "example_protos_closure",
|
|
visibility = ["//visibility:public"],
|
|
deps = [":example_protos"],
|
|
)
|
|
|
|
filegroup(
|
|
name = "mobile_srcs_no_runtime",
|
|
srcs = [
|
|
"feature_util.cc",
|
|
"feature_util.h",
|
|
],
|
|
)
|
|
|
|
# TODO(bmzhao): These files are exported since they are directly referenced
|
|
# from targets in tensorflow/core/BUILD. Remove these after phase 4 of the
|
|
# TF refactor: https://github.com/tensorflow/community/pull/179
|
|
exports_files(
|
|
srcs = [
|
|
"example.proto",
|
|
"example_parser_configuration_test.cc",
|
|
"feature.proto",
|
|
"feature_util.cc",
|
|
"feature_util.h",
|
|
"feature_util_test.cc",
|
|
],
|
|
visibility = [
|
|
"//tensorflow/core:__pkg__",
|
|
],
|
|
)
|