# Description: # TensorFlow is a computational framework, primarily for use in machine # learning applications. package( default_visibility = ["//visibility:public"], ) licenses(["notice"]) # Apache 2.0 load( "//tensorflow:tensorflow.bzl", "cc_library_with_android_deps", "tf_cc_binary", "tf_cc_test", "tf_copts", "tf_gen_op_wrappers_cc", "transitive_hdrs", ) 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", ":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: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", "framework/scope_internal.h", ], 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", ], ) 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/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_with_android_deps( name = "while_loop", srcs = ["ops/while_loop.cc"], hdrs = ["ops/while_loop.h"], android_deps = [ "//tensorflow/core:android_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", ], ) tf_gen_op_wrappers_cc( name = "cc_ops", api_def_srcs = ["//tensorflow/core/api_def:base_api_def"], op_lib_names = [ "array_ops", "audio_ops", "candidate_sampling_ops", "control_flow_ops", "data_flow_ops", "image_ops", "io_ops", "linalg_ops", "logging_ops", "lookup_ops", "manip_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", ], 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"], ) 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", visibility = ["//tensorflow:internal"], ) tf_gen_op_wrappers_cc( name = "remote_fused_graph_ops", op_lib_names = [ "remote_fused_graph_ops", ], pkg = "//tensorflow/core", ) 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(), 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:proto_text", "//tensorflow/core:protos_all_cc", ], ) 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:proto_text", "//tensorflow/core:protos_all_cc", "//tensorflow/core:test", "//tensorflow/core:test_main", ], ) 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:darwin": [ "-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", ], )