diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-28 14:08:25 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-28 14:34:43 -0700 |
commit | f83da5b0aa37ba55c1b2eaa093e6d043b73f5982 (patch) | |
tree | d28c727251f910dc0c7b7a6184286919d436e88f /tensorflow/core/BUILD | |
parent | 1724d155f00b49bc817189247cbfb0df2092a9da (diff) |
Introduce the abstraction of RunHandler which each DirectSession can use for
the duration of a single RunInternal() call from RunHandlerPool. It is used for
running inter-op closures with a global scheduler (which in the future) to
improve both median and tail latency (for use-cases like CPU inference).
In the case that global pools aren't used, this change should be a no-op.
PiperOrigin-RevId: 214992852
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 50fe308b73..7da4b9fbd0 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -2486,6 +2486,8 @@ FRAMEWORK_INTERNAL_PUBLIC_HEADERS = [ "framework/op_segment.h", "framework/rendezvous.h", # only needed for tests "framework/resource_var.h", + "framework/run_handler.h", + "framework/run_handler_util.h", "framework/tensor_reference.h", "framework/tracking_allocator.h", # only needed for tests "framework/unique_tensor_references.h", @@ -2972,6 +2974,7 @@ tf_cuda_library( ":core_cpu_internal", ":device_tracer", ":framework", + ":framework_internal", ":graph", ":lib", ":lib_internal", @@ -4119,6 +4122,19 @@ tf_cc_test( ], ) +tf_cc_test( + name = "framework_run_handler_util_test", + size = "small", + srcs = ["framework/run_handler_util_test.cc"], + linkstatic = tf_kernel_tests_linkstatic(), + deps = [ + ":framework_internal", + ":lib", + ":test", + ":test_main", + ], +) + tf_cuda_cc_test( name = "common_runtime_direct_session_test", size = "small", |