Skip to content

Commit 29ebda9

Browse files
committed
Update LMDBDataset to support batch at the creation
This PR updates LMDBDataset to support batch at the creation, it also switch to the new pattern to reduce unnecessary code duplication. Signed-off-by: Yong Tang <[email protected]>
1 parent 8242d26 commit 29ebda9

File tree

7 files changed

+195
-235
lines changed

7 files changed

+195
-235
lines changed

tensorflow_io/core/BUILD

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ cc_library(
3535
".",
3636
],
3737
deps = [
38+
"@libarchive",
3839
"@local_config_tf//:libtensorflow_framework",
3940
"@local_config_tf//:tf_header_lib",
4041
],

tensorflow_io/lmdb/BUILD

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,8 @@ package(default_visibility = ["//visibility:public"])
55
cc_binary(
66
name = "python/ops/_lmdb_ops.so",
77
srcs = [
8-
"kernels/lmdb_dataset_ops.cc",
9-
"ops/dataset_ops.cc",
8+
"kernels/lmdb_input.cc",
9+
"ops/lmdb_ops.cc",
1010
],
1111
copts = [
1212
"-pthread",
@@ -15,6 +15,7 @@ cc_binary(
1515
],
1616
linkshared = 1,
1717
deps = [
18+
"//tensorflow_io/core:dataset_ops",
1819
"@lmdb",
1920
"@local_config_tf//:libtensorflow_framework",
2021
"@local_config_tf//:tf_header_lib",

tensorflow_io/lmdb/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@
2121
from __future__ import division
2222
from __future__ import print_function
2323

24-
from tensorflow_io.lmdb.python.ops.lmdb_dataset_ops import LMDBDataset
24+
from tensorflow_io.lmdb.python.ops.lmdb_ops import LMDBDataset
2525

2626
from tensorflow.python.util.all_util import remove_undocumented
2727

tensorflow_io/lmdb/kernels/lmdb_dataset_ops.cc

Lines changed: 0 additions & 219 deletions
This file was deleted.

0 commit comments

Comments
 (0)