Skip to content

Commit eadf669

Browse files
ebrevdotensorflower-gardener
authored andcommitted
Added the PriorityQueue and Barrier TF objects.
Change: 126119692
1 parent 055fdda commit eadf669

13 files changed

Lines changed: 2879 additions & 0 deletions

File tree

tensorflow/core/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,7 @@ cc_library(
121121
"lib/gtl/array_slice.h",
122122
"lib/gtl/inlined_vector.h",
123123
"lib/gtl/map_util.h", # TODO(josh11b): make internal
124+
"lib/gtl/priority_queue_util.h",
124125
"lib/gtl/stl_util.h", # TODO(josh11b): make internal
125126
"lib/hash/crc32c.h", # TODO(josh11b): make internal
126127
"lib/hash/hash.h", # TODO(josh11b): make internal

tensorflow/core/kernels/BUILD

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -211,6 +211,19 @@ cc_library(
211211
],
212212
)
213213

214+
cc_library(
215+
name = "priority_queue",
216+
srcs = ["priority_queue.cc"],
217+
hdrs = ["priority_queue.h"],
218+
deps = [
219+
":queue_base",
220+
":queue_op",
221+
":typed_queue",
222+
"//tensorflow/core:framework",
223+
"//tensorflow/core:lib",
224+
],
225+
)
226+
214227
tf_proto_library(
215228
name = "reader_base_proto",
216229
srcs = ["reader_base.proto"],
@@ -675,7 +688,9 @@ tf_kernel_libraries(
675688
prefixes = [
676689
"dynamic_partition_op",
677690
"dynamic_stitch_op",
691+
"barrier_ops",
678692
"fifo_queue_op",
693+
"priority_queue_op",
679694
"lookup_table_init_op",
680695
"lookup_table_op",
681696
"padding_fifo_queue_op",
@@ -692,6 +707,7 @@ tf_kernel_libraries(
692707
":initializable_lookup_table",
693708
":lookup_util",
694709
":padding_fifo_queue",
710+
":priority_queue",
695711
":queue_base",
696712
":queue_op",
697713
":split_lib",

0 commit comments

Comments
 (0)