diff options
author | 2018-09-20 13:48:43 -0700 | |
---|---|---|
committer | 2018-09-20 13:53:34 -0700 | |
commit | d388770922ad1afa95e55597a33836fe74035c75 (patch) | |
tree | 3c15e17c357645e3e872ea46daaea2c91b00e9c1 /tensorflow/python/BUILD | |
parent | 1f1e5ac6154583d5f87c846d1d7c9c59a77d6e0c (diff) |
Implement TF graph capture.
PiperOrigin-RevId: 213875284
Diffstat (limited to 'tensorflow/python/BUILD')
-rw-r--r-- | tensorflow/python/BUILD | 15 |
1 files changed, 14 insertions, 1 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 9730e9933a..79f14466e6 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3763,6 +3763,19 @@ cuda_py_tests( ], ) +cc_library( + name = "session_ref", + srcs = ["client/session_ref.cc"], + hdrs = ["client/session_ref.h"], + deps = [ + "//tensorflow/core:core_cpu", + "//tensorflow/core:lib", + "//tensorflow/core:master_proto_cc", + "//tensorflow/core:protos_all_cc", + "//tensorflow/core:replay_log_proto_cc", + ], +) + tf_cuda_library( name = "tf_session_helper", srcs = ["client/tf_session_helper.cc"], @@ -3773,6 +3786,7 @@ tf_cuda_library( ":ndarray_tensor_bridge", ":numpy_lib", ":safe_ptr", + ":session_ref", ":test_ops_kernels", "//tensorflow/c:c_api", "//tensorflow/c:c_api_internal", @@ -3785,7 +3799,6 @@ tf_cuda_library( "//tensorflow/core:graph", "//tensorflow/core:lib", "//tensorflow/core:protos_all_cc", - "//tensorflow/core:session_ref", "//third_party/py/numpy:headers", "//third_party/python_runtime:headers", ], |