| load( |
| "//tensorflow/core/platform:build_config.bzl", |
| "tf_kernel_tests_linkstatic", |
| "tf_proto_library", |
| ) |
| load( |
| "//tensorflow/core/platform:rules_cc.bzl", |
| "cc_library", |
| ) |
| load( |
| "//tensorflow:tensorflow.bzl", |
| "tf_cc_test", |
| "tf_cc_tests", |
| "tf_copts", |
| "tf_cuda_library", |
| "tf_cuda_only_cc_test", |
| ) |
| load("//tensorflow:tensorflow.bzl", "tf_version_info_genrule") |
| load( |
| "//third_party/mkl:build_defs.bzl", |
| "mkl_deps", |
| ) |
| load( |
| "//tensorflow/core/platform:build_config_root.bzl", |
| "if_static", |
| ) |
| |
| package( |
| default_visibility = [ |
| "//tensorflow/core:__subpackages__", |
| ], |
| licenses = ["notice"], # Apache 2.0 |
| ) |
| |
| # List of exported proto source files. |
| exports_files( |
| srcs = [ |
| "event.proto", |
| "example_proto_fast_parsing_test.proto", |
| "memmapped_file_system.proto", |
| "saved_tensor_slice.proto", |
| ], |
| visibility = [ |
| "//tensorflow:internal", |
| "//tensorflow/core:__pkg__", |
| ], |
| ) |
| |
| # Export source files needed for mobile builds, which do not use granular targets. |
| filegroup( |
| name = "mobile_srcs_no_runtime", |
| srcs = [ |
| "overflow.h", |
| "padding.cc", |
| "padding.h", |
| "tensor_format.cc", |
| "tensor_format.h", |
| ], |
| ) |
| |
| filegroup( |
| name = "mobile_srcs_only_runtime", |
| srcs = [ |
| "batch_util.cc", |
| "batch_util.h", |
| "bcast.cc", |
| "bcast.h", |
| "command_line_flags.cc", |
| "command_line_flags.h", |
| "device_name_utils.cc", |
| "device_name_utils.h", |
| "dump_graph.cc", |
| "dump_graph.h", |
| "einsum_op_util.cc", |
| "einsum_op_util.h", |
| "env_var.cc", |
| "env_var.h", |
| "equal_graph_def.cc", |
| "equal_graph_def.h", |
| "example_proto_fast_parsing.cc", |
| "example_proto_fast_parsing.h", |
| "example_proto_helper.cc", |
| "example_proto_helper.h", |
| "guarded_philox_random.cc", |
| "guarded_philox_random.h", |
| "matmul_autotune.cc", |
| "matmul_autotune.h", |
| "matmul_bcast.cc", |
| "matmul_bcast.h", |
| "mirror_pad_mode.cc", |
| "mirror_pad_mode.h", |
| "port.cc", |
| "port.h", |
| "presized_cuckoo_map.h", |
| "ptr_util.h", |
| "reffed_status_callback.h", |
| "saved_tensor_slice_util.cc", |
| "saved_tensor_slice_util.h", |
| "stat_summarizer.cc", |
| "stat_summarizer.h", |
| "strided_slice_op.cc", |
| "strided_slice_op.h", |
| "tensor_ops_util.h", |
| "tensor_slice_reader.cc", |
| "tensor_slice_reader.h", |
| "tensor_slice_reader_cache.cc", |
| "tensor_slice_reader_cache.h", |
| "tensor_slice_set.cc", |
| "tensor_slice_set.h", |
| "tensor_slice_util.h", |
| "tensor_slice_writer.cc", |
| "tensor_slice_writer.h", |
| "use_cudnn.cc", |
| "use_cudnn.h", |
| "util.cc", |
| "util.h", |
| "work_sharder.cc", |
| "work_sharder.h", |
| ], |
| ) |
| |
| filegroup( |
| name = "framework_internal_private_hdrs", |
| srcs = [ |
| "activation_mode.h", |
| "batch_util.h", |
| "bcast.h", |
| "command_line_flags.h", |
| "debug_events_writer.h", |
| "device_name_utils.h", |
| "dump_graph.h", |
| "einsum_op_util.h", |
| "env_var.h", |
| "equal_graph_def.h", |
| "events_writer.h", |
| "example_proto_fast_parsing.h", |
| "example_proto_helper.h", |
| "exec_on_stall.h", |
| "gpu_cuda_alias.h", |
| "gpu_device_functions.h", |
| "gpu_kernel_helper.h", |
| "gpu_launch_config.h", |
| "guarded_philox_random.h", |
| "matmul_autotune.h", |
| "matmul_bcast.h", |
| "mirror_pad_mode.h", |
| "mkl_threadpool.h", |
| "mkl_types.h", |
| "mkl_util.h", |
| "overflow.h", |
| "padding.h", |
| "permutation_input_iterator.h", |
| "permutation_output_iterator.h", |
| "presized_cuckoo_map.h", |
| "ptr_util.h", |
| "reffed_status_callback.h", |
| "saved_tensor_slice_util.h", |
| "stat_summarizer.h", |
| "stat_summarizer_options.h", |
| "stats_calculator.h", |
| "stream_executor_util.h", |
| "strided_slice_op.h", |
| "tensor_format.h", |
| "tensor_ops_util.h", |
| "tensor_slice_reader.h", |
| "tensor_slice_reader_cache.h", |
| "tensor_slice_set.h", |
| "tensor_slice_util.h", |
| "tensor_slice_writer.h", |
| "transform_output_iterator.h", |
| "use_cudnn.h", |
| "util.h", |
| "work_sharder.h", |
| "xla_config_registry.h", |
| ], |
| ) |
| |
| filegroup( |
| name = "framework_internal_public_hdrs", |
| srcs = [ |
| "command_line_flags.h", |
| "equal_graph_def.h", |
| "presized_cuckoo_map.h", |
| "tensor_slice_set.h", |
| "tensor_slice_util.h", |
| "xla_config_registry.h", |
| ], |
| ) |
| |
| filegroup( |
| name = "framework_internal_impl_srcs", |
| srcs = [ |
| "activation_mode.cc", |
| "batch_util.cc", |
| "bcast.cc", |
| "command_line_flags.cc", |
| "debug_events_writer.cc", |
| "device_name_utils.cc", |
| "dump_graph.cc", |
| "equal_graph_def.cc", |
| "events_writer.cc", |
| "example_proto_fast_parsing.cc", |
| "example_proto_helper.cc", |
| "guarded_philox_random.cc", |
| "matmul_autotune.cc", |
| "matmul_bcast.cc", |
| "mirror_pad_mode.cc", |
| "saved_tensor_slice_util.cc", |
| "stat_summarizer.cc", |
| "strided_slice_op.cc", |
| "tensor_slice_reader.cc", |
| "tensor_slice_reader_cache.cc", |
| "tensor_slice_set.cc", |
| "tensor_slice_writer.cc", |
| "use_cudnn.cc", |
| "util.cc", |
| "work_sharder.cc", |
| "xla_config_registry.cc", |
| ], |
| ) |
| |
| filegroup( |
| name = "lib_internal_public_hdrs", |
| srcs = [ |
| "env_var.h", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "lib_hdrs", |
| srcs = [ |
| "gpu_cuda_alias.h", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "memmapped_file_system_hdrs", |
| srcs = [ |
| "memmapped_file_system.h", |
| "memmapped_file_system_writer.h", |
| ], |
| ) |
| |
| filegroup( |
| name = "memmapped_file_system_srcs", |
| srcs = [ |
| "memmapped_file_system.cc", |
| "memmapped_file_system_writer.cc", |
| ], |
| ) |
| |
| filegroup( |
| name = "port_hdrs", |
| srcs = [ |
| "port.h", |
| ], |
| visibility = [ |
| "//tensorflow/core:__pkg__", |
| "//tensorflow/python:__pkg__", |
| ], |
| ) |
| |
| filegroup( |
| name = "test_hdrs", |
| testonly = 1, |
| srcs = [ |
| "reporter.h", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "mkl_util_hdrs", |
| srcs = [ |
| "mkl_threadpool.h", |
| "mkl_util.h", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "mkl_util_test_srcs", |
| testonly = 1, |
| srcs = [ |
| "mkl_util_test.cc", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "android_test_hdrs", |
| testonly = 1, |
| srcs = [ |
| "reporter.h", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "android_test_srcs", |
| testonly = 1, |
| srcs = [ |
| "reporter.cc", |
| ":android_test_hdrs", |
| ], |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| filegroup( |
| name = "framework_srcs", |
| srcs = [ |
| "activation_mode.h", |
| "batch_util.h", |
| "bcast.h", |
| "debug_events_writer.h", |
| "device_name_utils.h", |
| "dump_graph.h", |
| "einsum_op_util.h", |
| "events_writer.h", |
| "example_proto_fast_parsing.h", |
| "example_proto_helper.h", |
| "gpu_kernel_helper.h", |
| "guarded_philox_random.h", |
| "matmul_autotune.h", |
| "matmul_bcast.h", |
| "mirror_pad_mode.h", |
| "padding.h", |
| "port.h", |
| "ptr_util.h", |
| "reffed_status_callback.h", |
| "saved_tensor_slice_util.h", |
| "stat_summarizer.h", |
| "stat_summarizer_options.h", |
| "stream_executor_util.h", |
| "strided_slice_op.h", |
| "tensor_format.h", |
| "tensor_ops_util.h", |
| "tensor_slice_reader.h", |
| "tensor_slice_reader_cache.h", |
| "tensor_slice_writer.h", |
| "use_cudnn.h", |
| "util.h", |
| "work_sharder.h", |
| ], |
| ) |
| |
| # Version info generation needs to be generated in the same package where it |
| # is written. |
| tf_version_info_genrule( |
| name = "version_info_gen", |
| out = "version_info.cc", |
| ) |
| |
| cc_library( |
| name = "stats_calculator_portable", |
| srcs = [ |
| "stat_summarizer_options.h", |
| "stats_calculator.cc", |
| ], |
| hdrs = [ |
| "stats_calculator.h", |
| ], |
| copts = tf_copts(), |
| visibility = [ |
| "//tensorflow:internal", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "stats_calculator_test", |
| srcs = ["stats_calculator_test.cc"], |
| deps = [ |
| ":stats_calculator_portable", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| cc_library( |
| name = "reporter", |
| srcs = ["reporter.cc"], |
| hdrs = ["reporter.h"], |
| visibility = ["//tensorflow/core:__subpackages__"], |
| deps = [ |
| ":test_log_proto_impl_cc", |
| "//tensorflow/core/platform:env", |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/core/platform:macros", |
| "//tensorflow/core/platform:mutex", |
| "//tensorflow/core/platform:str_util", |
| "//tensorflow/core/platform:types", |
| ], |
| ) |
| |
| cc_library( |
| name = "port", |
| srcs = ["port.cc"], |
| hdrs = ["port.h"], |
| copts = tf_copts(), |
| visibility = [ |
| "//tensorflow/core:__pkg__", |
| "//tensorflow/python:__pkg__", |
| ], |
| alwayslink = 1, |
| ) |
| |
| tf_cuda_library( |
| name = "gpu_cuda_alias", |
| hdrs = ["gpu_cuda_alias.h"], |
| ) |
| |
| tf_cuda_library( |
| name = "gpu_device_functions", |
| hdrs = ["gpu_device_functions.h"], |
| deps = [ |
| ":gpu_cuda_alias", |
| "//tensorflow/core/platform:types", |
| "//third_party/eigen3", |
| ], |
| ) |
| |
| cc_library( |
| name = "overflow", |
| hdrs = ["overflow.h"], |
| deps = [ |
| "//tensorflow/core/platform:logging", |
| "//tensorflow/core/platform:macros", |
| "//tensorflow/core/platform:types", |
| ], |
| ) |
| |
| cc_library( |
| name = "exec_on_stall", |
| hdrs = ["exec_on_stall.h"], |
| deps = [ |
| "//tensorflow/core/platform:env", |
| "//tensorflow/core/platform:mutex", |
| ], |
| ) |
| |
| cc_library( |
| name = "ptr_util", |
| hdrs = ["ptr_util.h"], |
| ) |
| |
| cc_library( |
| name = "version_info", |
| srcs = ["version_info.cc"], |
| hdrs = ["//tensorflow/core/public:version.h"], |
| copts = tf_copts(), |
| alwayslink = if_static(0, 1), |
| ) |
| |
| cc_library( |
| name = "tensor_format", |
| srcs = ["tensor_format.cc"], |
| hdrs = ["tensor_format.h"], |
| deps = [ |
| "//tensorflow/core/framework:tensor", |
| "//tensorflow/core/lib/gtl:array_slice", |
| "//tensorflow/core/lib/gtl:inlined_vector", |
| "//tensorflow/core/platform:types", |
| ], |
| ) |
| |
| cc_library( |
| name = "padding", |
| srcs = ["padding.cc"], |
| hdrs = ["padding.h"], |
| deps = [ |
| ":tensor_format", |
| "//tensorflow/core/framework:attr_value_proto_cc", |
| "//tensorflow/core/lib/core:errors", |
| "//tensorflow/core/lib/core:status", |
| ], |
| ) |
| |
| cc_library( |
| name = "einsum_op_util", |
| srcs = ["einsum_op_util.cc"], |
| hdrs = ["einsum_op_util.h"], |
| deps = [ |
| "//tensorflow/core/lib/core:errors", |
| "//tensorflow/core/lib/core:status", |
| "//tensorflow/core/lib/gtl:inlined_vector", |
| "@com_google_absl//absl/strings", |
| ], |
| ) |
| |
| cc_library( |
| name = "env_var", |
| srcs = ["env_var.cc"], |
| hdrs = ["env_var.h"], |
| deps = [ |
| "//tensorflow/core/platform:errors", |
| "//tensorflow/core/platform:logging", |
| "//tensorflow/core/platform:numbers", |
| "//tensorflow/core/platform:status", |
| "//tensorflow/core/platform:str_util", |
| "//tensorflow/core/platform:strcat", |
| "//tensorflow/core/platform:stringpiece", |
| "//tensorflow/core/platform:types", |
| ], |
| ) |
| |
| # Tests. |
| |
| tf_cc_test( |
| name = "overflow_test", |
| size = "small", |
| srcs = ["overflow_test.cc"], |
| deps = [ |
| ":overflow", |
| "//tensorflow/core:framework_lite", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "exec_on_stall_test", |
| size = "small", |
| srcs = ["exec_on_stall_test.cc"], |
| deps = [ |
| ":exec_on_stall", |
| "//tensorflow/core:framework_lite", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| ], |
| ) |
| |
| tf_cuda_only_cc_test( |
| name = "gpu_kernel_helper_test", |
| srcs = [ |
| "gpu_kernel_helper_test.cu.cc", |
| ], |
| deps = [ |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "//third_party/eigen3", |
| ] + mkl_deps(), |
| ) |
| |
| tf_cc_tests( |
| name = "higher_level_tests", |
| size = "small", |
| srcs = [ |
| "bcast_test.cc", |
| "command_line_flags_test.cc", |
| "device_name_utils_test.cc", |
| "dump_graph_test.cc", |
| "equal_graph_def_test.cc", |
| "events_writer_test.cc", |
| "example_proto_fast_parsing_test.cc", |
| "example_proto_helper_test.cc", |
| "matmul_bcast_test.cc", |
| "memmapped_file_system_test.cc", |
| "presized_cuckoo_map_test.cc", |
| "reffed_status_callback_test.cc", |
| "reporter_test.cc", |
| "saved_tensor_slice_util_test.cc", |
| "semver_test.cc", |
| "stat_summarizer_test.cc", |
| "tensor_format_test.cc", |
| "tensor_slice_reader_test.cc", |
| "tensor_slice_set_test.cc", |
| "tensor_slice_util_test.cc", |
| "tensor_slice_writer_test.cc", |
| "work_sharder_test.cc", |
| ], |
| create_named_test_suite = True, |
| linkopts = select({ |
| "//tensorflow:macos": ["-headerpad_max_install_names"], |
| "//conditions:default": [], |
| }), |
| linkstatic = tf_kernel_tests_linkstatic(), |
| visibility = [ |
| "//tensorflow/core:__pkg__", |
| ], |
| deps = [ |
| ":protos_test_cc", |
| "//tensorflow/cc:cc_ops", |
| "//tensorflow/cc:cc_ops_internal", |
| "//tensorflow/cc:function_ops", |
| "//tensorflow/cc:ops", |
| "//tensorflow/cc:scope", |
| "//tensorflow/cc:sendrecv_ops", |
| "//tensorflow/cc:while_loop", |
| "//tensorflow/core", |
| "//tensorflow/core:core_cpu", |
| "//tensorflow/core:core_cpu_internal", |
| "//tensorflow/core:direct_session_internal", |
| "//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:ops_util", |
| "//tensorflow/core/platform:regexp", |
| "//third_party/eigen3", |
| "@com_google_absl//absl/base", |
| "@com_google_absl//absl/container:flat_hash_set", |
| "@com_google_absl//absl/memory", |
| "@com_google_absl//absl/strings", |
| ], |
| ) |
| |
| tf_cc_test( |
| name = "debug_events_writer_test", |
| size = "small", |
| srcs = ["debug_events_writer_test.cc"], |
| linkopts = select({ |
| "//tensorflow:macos": ["-headerpad_max_install_names"], |
| "//conditions:default": [], |
| }), |
| linkstatic = tf_kernel_tests_linkstatic(), |
| tags = ["no_windows"], # b/150411480 |
| deps = [ |
| "//tensorflow/core:framework", |
| "//tensorflow/core:lib", |
| "//tensorflow/core:ops", |
| "//tensorflow/core:protos_all_cc", |
| "//tensorflow/core:test", |
| "//tensorflow/core:test_main", |
| "@com_google_absl//absl/container:flat_hash_set", |
| ], |
| ) |
| |
| # Proto libraries. |
| tf_proto_library( |
| name = "test_log_proto_impl", |
| srcs = ["test_log.proto"], |
| cc_api_version = 2, |
| make_default_target_header_only = True, |
| # Not to be used outside of tensorflow/core. |
| visibility = ["//tensorflow/core:__pkg__"], |
| ) |
| |
| tf_proto_library( |
| name = "protos_test", |
| srcs = ["example_proto_fast_parsing_test.proto"], |
| cc_api_version = 2, |
| protodeps = ["//tensorflow/core:protos_all"], |
| visibility = ["//visibility:public"], |
| ) |
| |
| tf_proto_library( |
| name = "event_proto", |
| srcs = ["event.proto"], |
| cc_api_version = 2, |
| make_default_target_header_only = True, |
| protodeps = [ |
| "//tensorflow/core/framework:summary_proto", |
| "//tensorflow/core/framework:resource_handle_proto", |
| "//tensorflow/core/framework:tensor_proto", |
| "//tensorflow/core/framework:tensor_shape_proto", |
| "//tensorflow/core/framework:types_proto", |
| ], |
| visibility = ["//visibility:public"], |
| ) |
| |
| tf_proto_library( |
| name = "saved_tensor_slice_proto", |
| srcs = ["saved_tensor_slice.proto"], |
| cc_api_version = 2, |
| make_default_target_header_only = True, |
| protodeps = [ |
| "//tensorflow/core/framework:resource_handle_proto", |
| "//tensorflow/core/framework:tensor_proto", |
| "//tensorflow/core/framework:tensor_shape_proto", |
| "//tensorflow/core/framework:tensor_slice_proto", |
| "//tensorflow/core/framework:types_proto", |
| "//tensorflow/core/framework:versions_proto", |
| ], |
| visibility = ["//visibility:public"], |
| ) |
| |
| tf_proto_library( |
| name = "memmapped_file_system_proto", |
| srcs = ["memmapped_file_system.proto"], |
| cc_api_version = 2, |
| make_default_target_header_only = True, |
| visibility = ["//visibility:public"], |
| ) |
| |
| tf_proto_library( |
| name = "protos_all", |
| cc_api_version = 2, |
| make_default_target_header_only = True, |
| protodeps = [ |
| ":event_proto", |
| ":saved_tensor_slice_proto", |
| ":memmapped_file_system_proto", |
| ], |
| ) |
