885 lines
23 KiB
Python
885 lines
23 KiB
Python
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",
|
|
"tf_kernel_library",
|
|
)
|
|
|
|
# buildifier: disable=same-origin-load
|
|
load("//tensorflow:tensorflow.bzl", "filegroup")
|
|
|
|
# buildifier: disable=same-origin-load
|
|
load("//tensorflow:tensorflow.bzl", "get_compatible_with_portable")
|
|
|
|
# buildifier: disable=same-origin-load
|
|
load("//tensorflow:tensorflow.bzl", "tf_version_info_genrule")
|
|
load(
|
|
"//tensorflow/core/platform:build_config_root.bzl",
|
|
"if_static",
|
|
)
|
|
load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda")
|
|
load(
|
|
"@local_config_rocm//rocm:build_defs.bzl",
|
|
"if_rocm",
|
|
)
|
|
|
|
default_package_visibility = [
|
|
"//tensorflow/core:__subpackages__",
|
|
]
|
|
|
|
package(
|
|
default_visibility = default_package_visibility,
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
# List of exported source files.
|
|
exports_files(
|
|
srcs = [
|
|
"event.proto",
|
|
"example_proto_fast_parsing_test.proto",
|
|
"image_resizer_state.h",
|
|
"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",
|
|
"managed_stack_trace.h",
|
|
"matmul_autotune.cc",
|
|
"matmul_autotune.h",
|
|
"matmul_bcast.h",
|
|
"mirror_pad_mode.cc",
|
|
"mirror_pad_mode.h",
|
|
"port.cc",
|
|
"port.h",
|
|
"presized_cuckoo_map.h",
|
|
"ptr_util.h",
|
|
"ragged_to_dense_util.cc",
|
|
"ragged_to_dense_util.h",
|
|
"ragged_to_dense_util_common.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",
|
|
"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__",
|
|
"//tensorflow/python/util:__pkg__",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "test_hdrs",
|
|
testonly = 1,
|
|
srcs = [
|
|
"reporter.h",
|
|
],
|
|
visibility = ["//tensorflow/core:__pkg__"],
|
|
)
|
|
|
|
filegroup(
|
|
name = "mkl_util_hdrs",
|
|
srcs = [
|
|
"mkl_threadpool.h",
|
|
"mkl_types.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",
|
|
"managed_stack_trace.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",
|
|
compatible_with = get_compatible_with_portable(),
|
|
)
|
|
|
|
cc_library(
|
|
name = "stats_calculator_portable",
|
|
srcs = [
|
|
"stat_summarizer_options.h",
|
|
"stats_calculator.cc",
|
|
],
|
|
hdrs = [
|
|
"stats_calculator.h",
|
|
],
|
|
copts = tf_copts(),
|
|
visibility = [
|
|
"//tensorflow:internal",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "ragged_to_dense_util_common",
|
|
hdrs = [
|
|
"ragged_to_dense_util_common.h",
|
|
],
|
|
visibility = [
|
|
"//visibility:public",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "ragged_to_dense_util",
|
|
srcs = [
|
|
"ragged_to_dense_util.cc",
|
|
],
|
|
hdrs = [
|
|
"ragged_to_dense_util.h",
|
|
],
|
|
deps = [
|
|
":ragged_to_dense_util_common",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "ragged_to_dense_util_test",
|
|
srcs = [
|
|
"ragged_to_dense_util_test.cc",
|
|
],
|
|
deps = [
|
|
":ragged_to_dense_util",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:testlib",
|
|
"@com_google_googletest//:gtest_main",
|
|
],
|
|
)
|
|
|
|
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:env_impl",
|
|
"//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__",
|
|
"//tensorflow/python/util:__pkg__",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "managed_stack_trace",
|
|
hdrs = ["managed_stack_trace.h"],
|
|
visibility = [
|
|
"//tensorflow/c/eager:__pkg__",
|
|
"//tensorflow/core:__pkg__",
|
|
"//tensorflow/core/common_runtime/eager:__pkg__",
|
|
"//tensorflow/core/framework:__pkg__",
|
|
"//tensorflow/core/platform:__pkg__",
|
|
"//tensorflow/python:__pkg__",
|
|
"//tensorflow/python/eager:__pkg__",
|
|
"//tensorflow/python/util:__pkg__",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core/platform:stack_frame",
|
|
"@com_google_absl//absl/container:flat_hash_map",
|
|
"@com_google_absl//absl/strings",
|
|
"@com_google_absl//absl/types:optional",
|
|
],
|
|
)
|
|
|
|
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"],
|
|
compatible_with = get_compatible_with_portable(),
|
|
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",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "incremental_barrier",
|
|
srcs = ["incremental_barrier.cc"],
|
|
hdrs = ["incremental_barrier.h"],
|
|
visibility = default_package_visibility + ["//tensorflow_serving:__subpackages__"],
|
|
deps = [
|
|
"//tensorflow/core:lib",
|
|
"@com_google_absl//absl/functional:bind_front",
|
|
],
|
|
)
|
|
|
|
tf_kernel_library(
|
|
name = "cuda_solvers",
|
|
srcs = ["cuda_solvers.cc"],
|
|
hdrs = ["cuda_solvers.h"],
|
|
compatible_with = [],
|
|
# @local_config_cuda//cuda:cusolver_static, //third_party/eigen3:blas,
|
|
# and //third_party/libf2c all contain various parts of BLAS, LAPACK,
|
|
# and f2c helper functions in global namespace. Tell the compiler to
|
|
# allow multiple definitions when linking this.
|
|
linkopts = select({
|
|
"//tensorflow:macos": [],
|
|
"//tensorflow:windows": [],
|
|
"//conditions:default": ["-Wl,-z,muldefs"],
|
|
}),
|
|
visibility = ["//tensorflow/core/kernels:friends"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core/platform/default/build_config:cublas_plugin",
|
|
"//tensorflow/stream_executor/cuda:cublas_lib",
|
|
"//tensorflow/stream_executor/cuda:cusolver_lib",
|
|
],
|
|
)
|
|
|
|
tf_kernel_library(
|
|
name = "rocm_solvers",
|
|
srcs = ["rocm_solvers.cc"],
|
|
hdrs = ["rocm_solvers.h"],
|
|
compatible_with = [],
|
|
visibility = ["//tensorflow/core/kernels:friends"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/stream_executor/lib",
|
|
"//tensorflow/stream_executor/platform:dso_loader",
|
|
"//tensorflow/stream_executor/rocm:rocblas_plugin",
|
|
"//tensorflow/stream_executor/rocm:rocm_gpu_executor",
|
|
] + if_rocm([
|
|
"@local_config_rocm//rocm:rocprim",
|
|
]),
|
|
)
|
|
|
|
tf_kernel_library(
|
|
name = "cuda_sparse",
|
|
srcs = if_cuda(["cuda_sparse.cc"]) + if_rocm(["rocm_sparse.cc"]),
|
|
hdrs = ["cuda_sparse.h"],
|
|
compatible_with = [],
|
|
deps = [
|
|
":cuda_solvers",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
] + if_cuda([
|
|
"//tensorflow/stream_executor/cuda:cusparse_lib",
|
|
"@local_config_cuda//cuda:cub_headers",
|
|
]) + if_rocm([
|
|
"//tensorflow/stream_executor/rocm:hipsparse_wrapper",
|
|
]),
|
|
)
|
|
|
|
cc_library(
|
|
name = "image_resizer_state",
|
|
hdrs = ["image_resizer_state.h"],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core/framework:bounds_check",
|
|
"//third_party/eigen3",
|
|
],
|
|
)
|
|
|
|
# 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",
|
|
],
|
|
tags = [
|
|
"no_cuda_asan", # TODO(b/171342366): re-enable.
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//third_party/eigen3",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
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: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/common_runtime:direct_session_internal",
|
|
"//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(),
|
|
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",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "incremental_barrier_test",
|
|
srcs = ["incremental_barrier_test.cc"],
|
|
deps = [
|
|
":incremental_barrier",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core/platform",
|
|
"@com_google_absl//absl/functional:bind_front",
|
|
"@com_google_absl//absl/time",
|
|
],
|
|
)
|
|
|
|
# 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",
|
|
],
|
|
tags = [
|
|
"alt_dep=//third_party/tensorflow/core:protos_all",
|
|
],
|
|
)
|