diff options
Diffstat (limited to 'tensorflow/core/ops/io_ops.cc')
-rw-r--r-- | tensorflow/core/ops/io_ops.cc | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/tensorflow/core/ops/io_ops.cc b/tensorflow/core/ops/io_ops.cc index f55a2e8560..fa12816c92 100644 --- a/tensorflow/core/ops/io_ops.cc +++ b/tensorflow/core/ops/io_ops.cc @@ -527,6 +527,21 @@ shared_name: If non-empty, this reader is named in the given bucket with this shared_name. Otherwise, the node name is used instead. )doc"); +REGISTER_OP("LMDBReader") + .Output("reader_handle: Ref(string)") + .Attr("container: string = ''") + .Attr("shared_name: string = ''") + .SetIsStateful() + .SetShapeFn(TwoElementOutput) + .Doc(R"doc( +A Reader that outputs the records from a LMDB file. +reader_handle: The handle to reference the Reader. +container: If non-empty, this reader is placed in the given container. + Otherwise, a default container is used. +shared_name: If non-empty, this reader is named in the given bucket + with this shared_name. Otherwise, the node name is used instead. +)doc"); + // TODO(cwhipkey): mark this deprecated in favor of V2. REGISTER_OP("IdentityReader") .Output("reader_handle: Ref(string)") |