d72465ac66
functions for C++. A souped-up version of the hidden_ops mechanism in Python, the intent is to use this for most or all of the client languages, with a common list of changes to make in a common file and per-language overrides. Also: * include the documentation for outputs in the generated comments * several updates to C++ API to match Python * fix C++ shape function for ConcatV2 now that we use it by default * split op_gen_lib out of core:framework, since it is only used by the op generators, and I don't want to add another proto to mobile builds Change: 146267344
531 lines
12 KiB
Python
531 lines
12 KiB
Python
# Description:
|
|
# TensorFlow is a computational framework, primarily for use in machine
|
|
# learning applications.
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
)
|
|
|
|
licenses(["notice"]) # Apache 2.0
|
|
|
|
exports_files(["LICENSE"])
|
|
|
|
load(
|
|
"//tensorflow:tensorflow.bzl",
|
|
"tf_cc_test",
|
|
"tf_copts",
|
|
"tf_gen_op_wrappers_cc",
|
|
"cc_library_with_android_deps",
|
|
)
|
|
|
|
cc_library(
|
|
name = "gradients",
|
|
srcs = ["framework/gradients.cc"],
|
|
hdrs = ["framework/gradients.h"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":ops",
|
|
":scope",
|
|
"//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",
|
|
":grad_op_registry",
|
|
":grad_ops",
|
|
":gradients",
|
|
":testutil",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//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",
|
|
":grad_op_registry",
|
|
":gradients",
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//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:core_cpu_internal",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library(
|
|
name = "grad_ops",
|
|
deps = [
|
|
":array_grad",
|
|
":math_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:android_tensorflow_lib"],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "scope",
|
|
srcs = ["framework/scope.cc"],
|
|
hdrs = ["framework/scope.h"],
|
|
android_deps = ["//tensorflow/core:android_tensorflow_lib"],
|
|
common_deps = [
|
|
":ops",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "framework_scope_test",
|
|
srcs = ["framework/scope_test.cc"],
|
|
deps = [
|
|
":ops",
|
|
":scope",
|
|
"//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:android_tensorflow_lib"],
|
|
common_deps = [
|
|
":ops",
|
|
":scope",
|
|
],
|
|
deps = [
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:lib",
|
|
"//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:tensorflow",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "const_op",
|
|
srcs = ["ops/const_op.cc"],
|
|
hdrs = ["ops/const_op.h"],
|
|
android_deps = [
|
|
"//tensorflow/core:android_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(
|
|
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",
|
|
],
|
|
)
|
|
|
|
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 = "math_grad",
|
|
srcs = ["gradients/math_grad.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_math_grad_test",
|
|
srcs = ["gradients/math_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":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",
|
|
":grad_op_registry",
|
|
":ops",
|
|
":scope",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
],
|
|
)
|
|
|
|
tf_cc_test(
|
|
name = "gradients_nn_grad_test",
|
|
srcs = ["gradients/nn_grad_test.cc"],
|
|
deps = [
|
|
":cc_ops",
|
|
":grad_op_registry",
|
|
":grad_testutil",
|
|
":gradient_checker",
|
|
":nn_grad",
|
|
":testutil",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:test",
|
|
"//tensorflow/core:test_main",
|
|
"//tensorflow/core:testlib",
|
|
],
|
|
)
|
|
|
|
tf_gen_op_wrappers_cc(
|
|
name = "cc_ops",
|
|
op_lib_names = [
|
|
"array_ops",
|
|
"candidate_sampling_ops",
|
|
"control_flow_ops",
|
|
"data_flow_ops",
|
|
"image_ops",
|
|
"io_ops",
|
|
"linalg_ops",
|
|
"logging_ops",
|
|
"math_ops",
|
|
"nn_ops",
|
|
"no_op",
|
|
"parsing_ops",
|
|
"random_ops",
|
|
"sparse_ops",
|
|
"state_ops",
|
|
"string_ops",
|
|
"training_ops",
|
|
"user_ops",
|
|
],
|
|
other_hdrs = [
|
|
"ops/const_op.h",
|
|
"ops/standard_ops.h",
|
|
],
|
|
override_file = "ops/op_gen_overrides.pbtxt",
|
|
pkg = "//tensorflow/core",
|
|
)
|
|
|
|
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"],
|
|
)
|
|
|
|
cc_library_with_android_deps(
|
|
name = "cc_op_gen_main",
|
|
srcs = [
|
|
"framework/cc_op_gen.cc",
|
|
"framework/cc_op_gen.h",
|
|
"framework/cc_op_gen_main.cc",
|
|
],
|
|
android_deps = [
|
|
"//tensorflow/core:android_tensorflow_lib",
|
|
],
|
|
copts = tf_copts(),
|
|
deps = [
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:op_gen_lib",
|
|
"//tensorflow/core:proto_text",
|
|
"//tensorflow/core:protos_all_cc",
|
|
],
|
|
)
|
|
|
|
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:all_kernels",
|
|
"//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",
|
|
],
|
|
)
|
|
|
|
cc_binary(
|
|
name = "tutorials_example_trainer",
|
|
srcs = ["tutorials/example_trainer.cc"],
|
|
copts = tf_copts(),
|
|
linkopts = [
|
|
"-lpthread",
|
|
"-lm",
|
|
],
|
|
deps = [
|
|
":cc_ops",
|
|
"//tensorflow/core:all_kernels",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
],
|
|
)
|
|
|
|
filegroup(
|
|
name = "all_files",
|
|
srcs = glob(
|
|
["**/*"],
|
|
exclude = [
|
|
"**/METADATA",
|
|
"**/OWNERS",
|
|
],
|
|
),
|
|
visibility = ["//tensorflow:__subpackages__"],
|
|
)
|
|
|
|
cc_library(
|
|
name = "queue_runner",
|
|
srcs = ["training/queue_runner.cc"],
|
|
hdrs = ["training/queue_runner.h"],
|
|
deps = [
|
|
":coordinator",
|
|
"//tensorflow/core:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
"//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:core_cpu",
|
|
"//tensorflow/core:framework",
|
|
"//tensorflow/core:lib",
|
|
"//tensorflow/core:lib_internal",
|
|
"//tensorflow/core:protos_all_cc",
|
|
"//tensorflow/core:tensorflow",
|
|
],
|
|
)
|
|
|
|
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",
|
|
],
|
|
)
|