Adding unbounded_work_queue to tensorflow/core/platform/BUILD.

PiperOrigin-RevId: 269446152
This commit is contained in:
Brian Zhao 2019-09-16 16:22:20 -07:00 committed by TensorFlower Gardener
parent 01427f1437
commit 5059b5968b
3 changed files with 28 additions and 1 deletions

View File

@ -68,6 +68,7 @@ tf_instantiate_platform_libraries(names = [
"strong_hash", "strong_hash",
"subprocess", "subprocess",
"tracing", "tracing",
"unbounded_work_queue",
"wide_char", "wide_char",
]) ])
@ -542,6 +543,15 @@ cc_library(
], ],
) )
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 # Below libraries are here only to make sure the legacy build rules
# in tensorflow/core/BUILD are working! # in tensorflow/core/BUILD are working!

View File

@ -279,6 +279,23 @@ TF_PLATFORM_LIBRARIES = {
"tags": ["no_oss", "manual"], "tags": ["no_oss", "manual"],
"visibility": ["//visibility:private"], "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 = { TF_WINDOWS_PLATFORM_LIBRARIES = {

View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
limitations under the License. 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 "absl/memory/memory.h"
#include "tensorflow/core/platform/env.h" #include "tensorflow/core/platform/env.h"