commit 5059b5968b6421d1ec36d0559522f572971fbedb
author Brian Zhao <bmzhao@google.com> Mon Sep 16 16:22:20 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Sep 16 17:07:27 2019 -0700
tree d57e5bbf366295ac6a4c6fabab8c6a2709bd0bd5
parent 01427f1437d08cf4193bc6c9c501425348e8ddd4

Adding unbounded_work_queue to tensorflow/core/platform/BUILD.

PiperOrigin-RevId: 269446152

tensorflow/core/platform/BUILD

diff --git a/tensorflow/core/platform/BUILD b/tensorflow/core/platform/BUILD
index 2ba0bc6..442a6d9 100644
--- a/tensorflow/core/platform/BUILD
+++ b/tensorflow/core/platform/BUILD
@@ -68,6 +68,7 @@
     "strong_hash",
     "subprocess",
     "tracing",
+    "unbounded_work_queue",
     "wide_char",
 ])
 
@@ -542,6 +543,15 @@
     ],
 )
 
+cc_library(
+    name = "unbounded_work_queue",
+    hdrs = ["unbounded_work_queue.h"],
+    deps = [
+        ":platform",
+        ":unbounded_work_queue_impl",
+    ],
+)
+
 # --------------------------------------------------------------------------
 #     Below libraries are here only to make sure the legacy build rules
 #     in tensorflow/core/BUILD are working!

tensorflow/core/platform/default/build_refactor.bzl

diff --git a/tensorflow/core/platform/default/build_refactor.bzl b/tensorflow/core/platform/default/build_refactor.bzl
index 7ef5002..fa119b0 100644
--- a/tensorflow/core/platform/default/build_refactor.bzl
+++ b/tensorflow/core/platform/default/build_refactor.bzl
@@ -279,6 +279,23 @@
         "tags": ["no_oss", "manual"],
         "visibility": ["//visibility:private"],
     },
+    "unbounded_work_queue": {
+        "name": "unbounded_work_queue_impl",
+        "hdrs": [
+            "//tensorflow/core/platform:default/unbounded_work_queue.h",
+        ],
+        "srcs": [
+            "//tensorflow/core/platform:default/unbounded_work_queue.cc",
+        ],
+        "deps": [
+            "@com_google_absl//absl/memory",
+            "//tensorflow/core/platform:env",
+            "//tensorflow/core/platform:mutex",
+            "//tensorflow/core/lib/core:notification",
+        ],
+        "tags": ["no_oss", "manual"],
+        "visibility": ["//visibility:private"],
+    },
 }
 
 TF_WINDOWS_PLATFORM_LIBRARIES = {

tensorflow/core/platform/default/unbounded_work_queue.cc

diff --git a/tensorflow/core/platform/default/unbounded_work_queue.cc b/tensorflow/core/platform/default/unbounded_work_queue.cc
index 3cc66b6..9e619ef 100644
--- a/tensorflow/core/platform/default/unbounded_work_queue.cc
+++ b/tensorflow/core/platform/default/unbounded_work_queue.cc
@@ -13,7 +13,7 @@
 limitations under the License.
 ==============================================================================*/
 
-#include "tensorflow/core/platform/unbounded_work_queue.h"
+#include "tensorflow/core/platform/default/unbounded_work_queue.h"
 
 #include "absl/memory/memory.h"
 #include "tensorflow/core/platform/env.h"