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",
"subprocess",
"tracing",
"unbounded_work_queue",
"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
# in tensorflow/core/BUILD are working!

View File

@ -279,6 +279,23 @@ TF_PLATFORM_LIBRARIES = {
"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 = {

View File

@ -13,7 +13,7 @@ See the License for the specific language governing permissions and
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"