diff options
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 18 |
1 files changed, 15 insertions, 3 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 79ad3b8e54..1a86bff5cd 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -168,6 +168,7 @@ COMMON_PROTO_SRCS = [ "example/example.proto", "example/feature.proto", "framework/allocation_description.proto", + "framework/api_def.proto", "framework/attr_value.proto", "framework/cost_graph.proto", "framework/device_attributes.proto", @@ -177,9 +178,9 @@ COMMON_PROTO_SRCS = [ "framework/iterator.proto", "framework/kernel_def.proto", "framework/log_memory.proto", + "framework/model.proto", "framework/node_def.proto", "framework/op_def.proto", - "framework/api_def.proto", "framework/reader_base.proto", "framework/remote_fused_graph_execute_info.proto", "framework/resource_handle.proto", @@ -299,6 +300,7 @@ filegroup( name = "platform_base_hdrs", srcs = [ "platform/byte_order.h", + "platform/cord.h", "platform/env_time.h", "platform/logging.h", "platform/macros.h", @@ -720,6 +722,7 @@ cc_library( name = "abi", srcs = ["platform/abi.cc"], hdrs = ["platform/abi.h"], + deps = [":platform_base"], ) cc_library( @@ -839,6 +842,7 @@ tf_cuda_library( "framework/log_memory.h", "framework/lookup_interface.h", "framework/memory_types.h", + "framework/model.h", "framework/node_def_builder.h", "framework/node_def_util.h", "framework/numeric_op.h", @@ -874,7 +878,6 @@ tf_cuda_library( "util/bcast.h", "util/cuda_kernel_helper.h", "util/device_name_utils.h", - "util/env_var.h", "util/events_writer.h", "util/example_proto_fast_parsing.h", "util/example_proto_helper.h", @@ -1918,6 +1921,13 @@ tf_pyclif_proto_library( ) tf_pyclif_proto_library( + name = "protobuf/config_pyclif", + proto_lib = ":protos_all_cc", + proto_srcfile = "protobuf/config.proto", + visibility = ["//visibility:public"], +) + +tf_pyclif_proto_library( name = "protobuf/device_properties_pyclif", proto_lib = ":protos_all_cc", proto_srcfile = "protobuf/device_properties.proto", @@ -2056,6 +2066,7 @@ LIB_INTERNAL_PUBLIC_HEADERS = tf_additional_lib_hdrs() + [ "platform/snappy.h", "platform/tensor_coding.h", "platform/tracing.h", + "util/env_var.h", ] # Replicated for lib_internal and lib_internal_impl. @@ -2095,6 +2106,7 @@ cc_library( "platform/*.cc", "platform/profile_utils/**/*.cc", "framework/resource_handle.cc", + "util/env_var.cc", ], exclude = [ "**/*test*", @@ -2450,7 +2462,6 @@ FRAMEWORK_INTERNAL_PUBLIC_HEADERS = [ "framework/unique_tensor_references.h", "framework/variant.h", "util/command_line_flags.h", - "util/env_var.h", "util/equal_graph_def.h", "util/presized_cuckoo_map.h", "util/tensor_slice_set.h", @@ -2526,6 +2537,7 @@ tf_cuda_library( "util/memmapped_file_system_writer.*", "util/stats_calculator.*", "util/version_info.cc", + "util/env_var.cc", ], ) + select({ "//tensorflow:windows": [], |