863 lines
20 KiB
Python
863 lines
20 KiB
Python
# Description:
|
|
# TensorFlow is a computational framework, primarily for use in machine
|
|
# learning applications.
|
|
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"cc_library_with_android_deps",
|
|
"tf_cc_binary",
|
|
"tf_cc_test",
|
|
"tf_copts",
|
|
"tf_gen_op_wrappers_cc",
|
|
"transitive_hdrs",
|
|
)
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
licenses = ["notice"], # Apache 2.0
|
|
)
|
|
|
|
filegroup(
|
|
name = "srcs_no_runtime",
|
|
srcs = [
|
|
"framework/gradients.h",
|
|
"framework/ops.h",
|
|
"framework/scope.h",
|
|
"framework/scope_internal.h",
|
|
"//tensorflow/cc/saved_model:loader.h",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "srcs",
|
|
srcs = [
|
|
"framework/gradients.h",
|
|
"framework/ops.h",
|
|
"framework/scope.h",
|
|
"framework/scope_internal.h",
|
|
"ops/array_ops.h",
|
|
"ops/while_loop.h",
|
|
"//tensorflow/cc/saved_model:loader.h",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "pywrap_required_hdrs",
|
|
srcs = [
|
|
"training/coordinator.h",
|
|
],
|
|
visibility = [
|
|
"//tensorflow/python:__pkg__",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "gradients",
|
|
srcs = [
|
|
"framework/gradients.cc",
|
|
"framework/while_gradients.cc",
|
|
"framework/while_gradients.h",
|
|
],
|
|
hdrs = ["framework/gradients.h"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":ops",
|
|
":scope",
|
|
":scope_internal",
|
|
":while_loop",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_gradients_test",
|
|
srcs = ["framework/gradients_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":grad_op_registry",
|
|
":grad_ops",
|
|
":gradients",
|
|
":testutil",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_while_gradients_test",
|
|
size = "small",
|
|
srcs = ["framework/while_gradients_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":grad_op_registry",
|
|
":grad_ops",
|
|
":gradients",
|
|
":testutil",
|
|
":while_loop",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "gradient_checker",
|
|
srcs = ["framework/gradient_checker.cc"],
|
|
hdrs = ["framework/gradient_checker.h"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":gradients",
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_gradient_checker_test",
|
|
srcs = ["framework/gradient_checker_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":grad_ops",
|
|
":gradient_checker",
|
|
":testutil",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:framework_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "grad_ops",
|
|
deps = [
|
|
":array_grad",
|
|
":data_flow_grad",
|
|
":image_grad",
|
|
":manip_grad",
|
|
":math_grad",
|
|
":nn_grad",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "grad_testutil",
|
|
testonly = 1,
|
|
srcs = ["gradients/grad_testutil.cc"],
|
|
hdrs = ["gradients/grad_testutil.h"],
|
|
deps = [
|
|
":grad_op_registry",
|
|
":ops",
|
|
":scope",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "ops",
|
|
srcs = ["framework/ops.cc"],
|
|
hdrs = ["framework/ops.h"],
|
|
android_deps = ["//tensorflow/core:portable_tensorflow_lib"],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:graph",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:ops",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "scope",
|
|
srcs = [
|
|
"framework/scope.cc",
|
|
"framework/scope_internal.h",
|
|
],
|
|
hdrs = ["framework/scope.h"],
|
|
android_deps = ["//tensorflow/core:portable_tensorflow_lib"],
|
|
common_deps = [
|
|
":ops",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "scope_internal",
|
|
hdrs = ["framework/scope_internal.h"],
|
|
common_deps = [
|
|
":scope",
|
|
],
|
|
deps = [],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_scope_test",
|
|
srcs = ["framework/scope_test.cc"],
|
|
deps = [
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/cc:cc_ops",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "client_session",
|
|
srcs = ["client/client_session.cc"],
|
|
hdrs = ["client/client_session.h"],
|
|
android_deps = ["//tensorflow/core:portable_tensorflow_lib"],
|
|
common_deps = [
|
|
":ops",
|
|
":scope",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_experimental",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "client_client_session_test",
|
|
srcs = ["client/client_session_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_experimental",
|
|
"//tensorflow/core:tensorflow",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
"//third_party/eigen3",
|
|
"@com_google_absl//absl/synchronization",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "const_op",
|
|
srcs = ["ops/const_op.cc"],
|
|
hdrs = ["ops/const_op.h"],
|
|
android_deps = [
|
|
"//tensorflow/core:portable_tensorflow_lib",
|
|
],
|
|
common_deps = [
|
|
":ops",
|
|
":scope",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "ops_const_op_test",
|
|
srcs = ["ops/const_op_test.cc"],
|
|
deps = [
|
|
":const_op",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "while_loop",
|
|
srcs = ["ops/while_loop.cc"],
|
|
hdrs = ["ops/while_loop.h"],
|
|
android_deps = [
|
|
"//tensorflow/core:portable_tensorflow_lib",
|
|
],
|
|
common_deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":ops",
|
|
":scope",
|
|
":scope_internal",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "ops_while_loop_test",
|
|
size = "small",
|
|
srcs = ["ops/while_loop_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":testutil",
|
|
":while_loop",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "grad_op_registry",
|
|
srcs = ["framework/grad_op_registry.cc"],
|
|
hdrs = ["framework/grad_op_registry.h"],
|
|
deps = [
|
|
":ops",
|
|
":scope",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "array_grad",
|
|
srcs = ["gradients/array_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
"//tensorflow/core:lib_proto_parsing",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_array_grad_test",
|
|
srcs = ["gradients/array_grad_test.cc"],
|
|
deps = [
|
|
":array_grad",
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":testutil",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "image_grad",
|
|
srcs = ["gradients/image_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_image_grad_test",
|
|
srcs = ["gradients/image_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":image_grad",
|
|
":testutil",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "math_grad",
|
|
srcs = ["gradients/math_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_math_grad_test",
|
|
srcs = ["gradients/math_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":gradients",
|
|
":math_grad",
|
|
":testutil",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "nn_grad",
|
|
srcs = ["gradients/nn_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_nn_grad_test",
|
|
srcs = ["gradients/nn_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":nn_grad",
|
|
":testutil",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "data_flow_grad",
|
|
srcs = ["gradients/data_flow_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":cc_ops_internal",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_data_flow_grad_test",
|
|
size = "small",
|
|
srcs = ["gradients/data_flow_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":data_flow_grad",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":testutil",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "manip_grad",
|
|
srcs = ["gradients/manip_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_manip_grad_test",
|
|
srcs = ["gradients/manip_grad_test.cc"],
|
|
deps = [
|
|
":array_ops",
|
|
":cc_ops",
|
|
":gradient_checker",
|
|
":manip_grad",
|
|
":testutil",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
# Generates separate libraries for array_ops and math_ops to reduce the dependency count of targets that depend on only these
|
|
tf_gen_op_wrappers_cc(
|
|
name = "math_ops",
|
|
api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
|
|
op_lib_names = [
|
|
"math_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "array_ops",
|
|
api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
|
|
extra_gen_deps = ["//tensorflow/c/kernels:bitcast_op_lib"],
|
|
op_lib_names = [
|
|
"array_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "cc_ops",
|
|
api_def_srcs = ["//tensorflow/core/api_def:base_api_def"],
|
|
deps_internal = [
|
|
":array_ops_internal",
|
|
":math_ops_internal",
|
|
],
|
|
op_lib_names = [
|
|
"audio_ops",
|
|
"candidate_sampling_ops",
|
|
"control_flow_ops",
|
|
"data_flow_ops",
|
|
"image_ops",
|
|
"io_ops",
|
|
"linalg_ops",
|
|
"list_ops",
|
|
"logging_ops",
|
|
"lookup_ops",
|
|
"manip_ops",
|
|
"nn_ops",
|
|
"no_op",
|
|
"parsing_ops",
|
|
"random_ops",
|
|
"sparse_ops",
|
|
"state_ops",
|
|
"string_ops",
|
|
"training_ops",
|
|
"user_ops",
|
|
],
|
|
other_hdrs = [
|
|
"ops/array_ops.h",
|
|
"ops/const_op.h",
|
|
"ops/math_ops.h",
|
|
"ops/standard_ops.h",
|
|
],
|
|
other_hdrs_internal = [
|
|
"ops/array_ops_internal.h",
|
|
"ops/math_ops_internal.h",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
deps = [
|
|
":array_ops",
|
|
":const_op",
|
|
":math_ops",
|
|
"//tensorflow/cc:ops",
|
|
"//tensorflow/cc:scope",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_cc_ops_test",
|
|
srcs = ["framework/cc_ops_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":test_op",
|
|
":test_op_op_lib",
|
|
":testutil",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "sendrecv_ops",
|
|
include_internal_ops = 1,
|
|
op_lib_names = [
|
|
"sendrecv_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "function_ops",
|
|
include_internal_ops = 1,
|
|
op_lib_names = [
|
|
"function_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
visibility = ["//tensorflow:internal"],
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "functional_ops",
|
|
include_internal_ops = 1,
|
|
op_lib_names = [
|
|
"functional_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
visibility = ["//tensorflow:internal"],
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "resource_variable_ops",
|
|
include_internal_ops = 1,
|
|
op_lib_names = [
|
|
"resource_variable_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "remote_fused_graph_ops",
|
|
op_lib_names = [
|
|
"remote_fused_graph_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "tpu_ops",
|
|
include_internal_ops = 1,
|
|
op_lib_names = [
|
|
"tpu_configuration_ops",
|
|
"tpu_cross_replica_ops",
|
|
"tpu_embedding_ops",
|
|
"tpu_embedding_load_retrieve_ops",
|
|
"tpu_functional_ops",
|
|
"tpu_heartbeat_ops",
|
|
"tpu_host_compute_ops",
|
|
"tpu_infeed_ops",
|
|
"tpu_outfeed_ops",
|
|
"tpu_ordinal_selector_ops",
|
|
"tpu_replication_ops",
|
|
],
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
cc_library(
|
|
name = "cc_op_gen_main",
|
|
srcs = [
|
|
"framework/cc_op_gen.cc",
|
|
"framework/cc_op_gen.h",
|
|
"framework/cc_op_gen_main.cc",
|
|
],
|
|
copts = tf_copts(),
|
|
data = [
|
|
"//tensorflow/core/api_def:base_api_def",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:framework_headers_lib",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:op_gen_lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "cc_op_gen_test",
|
|
srcs = [
|
|
"framework/cc_op_gen.cc",
|
|
"framework/cc_op_gen.h",
|
|
"framework/cc_op_gen_test.cc",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:op_gen_lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"@com_google_absl//absl/strings",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "test_op_op_lib",
|
|
srcs = ["framework/test_op.cc"],
|
|
linkstatic = 1,
|
|
deps = ["//tensorflow/core:framework"],
|
|
alwayslink = 1,
|
|
)
|
|
|
|
cc_library(
|
|
name = "testutil",
|
|
testonly = 1,
|
|
srcs = ["framework/testutil.cc"],
|
|
hdrs = ["framework/testutil.h"],
|
|
deps = [
|
|
":client_session",
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:tensorflow",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "test_op",
|
|
op_lib_names = [
|
|
"test_op",
|
|
],
|
|
)
|
|
|
|
tf_cc_binary(
|
|
name = "tutorials_example_trainer",
|
|
srcs = ["tutorials/example_trainer.cc"],
|
|
copts = tf_copts(),
|
|
linkopts = select({
|
|
"//tensorflow:windows": [],
|
|
"//tensorflow:macos": [
|
|
"-lm",
|
|
"-lpthread",
|
|
],
|
|
"//tensorflow:ios": [
|
|
"-lm",
|
|
"-lpthread",
|
|
],
|
|
"//conditions:default": [
|
|
"-lm",
|
|
"-lpthread",
|
|
"-lrt",
|
|
],
|
|
}),
|
|
deps = [
|
|
":cc_ops",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "queue_runner",
|
|
srcs = ["training/queue_runner.cc"],
|
|
hdrs = ["training/queue_runner.h"],
|
|
deps = [
|
|
":coordinator",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core/kernels:ops_util",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "queue_runner_test",
|
|
srcs = ["training/queue_runner_test.cc"],
|
|
deps = [
|
|
"coordinator",
|
|
":cc_ops",
|
|
":queue_runner",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "coordinator",
|
|
srcs = ["training/coordinator.cc"],
|
|
hdrs = ["training/coordinator.h"],
|
|
deps = [
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "coordinator_test",
|
|
srcs = ["training/coordinator_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":coordinator",
|
|
":queue_runner",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
transitive_hdrs(
|
|
name = "headers",
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
deps = [
|
|
":cc_ops",
|
|
":client_session",
|
|
":coordinator",
|
|
":gradient_checker",
|
|
":gradients",
|
|
":ops",
|
|
":queue_runner",
|
|
":remote_fused_graph_ops",
|
|
":scope",
|
|
"//tensorflow/cc/profiler",
|
|
"//tensorflow/cc/saved_model:constants",
|
|
"//tensorflow/cc/saved_model:loader",
|
|
"//tensorflow/cc/saved_model:reader",
|
|
"//tensorflow/cc/saved_model:signature_constants",
|
|
"//tensorflow/cc/saved_model:tag_constants",
|
|
"//tensorflow/cc/tools:freeze_saved_model",
|
|
],
|
|
)
|