From 6ee77b0b8e882dc9c419d89701269f41f438a379 Mon Sep 17 00:00:00 2001
From: Mingming Liu <mingmingl@google.com>
Date: Sat, 18 Jul 2020 16:29:23 -0700
Subject: [PATCH] Move helper class ThreadSafeStatus into a separate file with
unit test.
PiperOrigin-RevId: 321974213
Change-Id: I78ceb91618c40da799097aa4d2048be0bf182c16
---
tensorflow/core/kernels/batching_util/BUILD | 12 ++++++++++++
1 file changed, 12 insertions(+)
diff --git a/tensorflow/core/kernels/batching_util/BUILD b/tensorflow/core/kernels/batching_util/BUILD
index a23857cc8ce..ec246323679 100644
--- a/tensorflow/core/kernels/batching_util/BUILD
+++ b/tensorflow/core/kernels/batching_util/BUILD
@@ -64,6 +64,18 @@ cc_library(
],
)
+cc_library(
+ name = "threadsafe_status",
+ srcs = ["threadsafe_status.cc"],
+ hdrs = ["threadsafe_status.h"],
+ deps = [
+ "//tensorflow/core:lib",
+ "@com_google_absl//absl/base:core_headers",
+ "@com_google_absl//absl/status",
+ "@com_google_absl//absl/synchronization",
+ ],
+)
+
tf_cc_test(
name = "batch_scheduler_test",
srcs = ["batch_scheduler_test.cc"],