aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/batching
diff options
context:
space:
mode:
authorGravatar Christopher Olston <olston@google.com>2017-04-27 12:12:21 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-04-27 13:24:43 -0700
commit421aaddb0785a05be5443a78ae480aff6050e4f4 (patch)
tree4f3a85921ddc7bbd4a54b5afb02b17fedc40fdb3 /tensorflow/contrib/batching
parentb8e4c4485894daa6e4f2bb3206428cb5421689f3 (diff)
Add headers to targets in tensorflow/contrib/batching/BUILD.
Change: 154460280
Diffstat (limited to 'tensorflow/contrib/batching')
-rw-r--r--tensorflow/contrib/batching/BUILD2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/batching/BUILD b/tensorflow/contrib/batching/BUILD
index 93c00b1f0b..1f1e2db69b 100644
--- a/tensorflow/contrib/batching/BUILD
+++ b/tensorflow/contrib/batching/BUILD
@@ -19,6 +19,7 @@ cc_library(
cc_library(
name = "batch_scheduler",
+ hdrs = ["batch_scheduler.h"],
deps = [
":batch_scheduler_hdrs",
"//tensorflow/core:lib",
@@ -49,6 +50,7 @@ cc_library(
cc_library(
name = "shared_batch_scheduler",
+ hdrs = ["shared_batch_scheduler.h"],
deps = [
":batch_scheduler",
":shared_batch_scheduler_hdrs",