Rework on HDF5: add list_hdf5_datasets and read_hdf5 ops

tensorflow_io/core/python/ops/data_ops.py

@@ -58,9 +58,8 @@ def _apply_fn(dataset):
 class BaseDataset(tf.compat.v2.data.Dataset):
   """A Base Dataset"""
 
-  def __init__(self, variant, batch, dtypes, shapes):
+  def __init__(self, variant, dtypes, shapes):
     """Create a Base Dataset."""
-    self._batch = 0 if batch is None else batch
     self._dtypes = dtypes
     self._shapes = shapes
     super(BaseDataset, self).__init__(variant)
@@ -93,4 +92,4 @@ def __init__(self, fn, data_input, batch, dtypes, shapes):
         self._data_input,
         self._batch,
         output_types=self._dtypes,
-        output_shapes=self._shapes), self._batch, self._dtypes, self._shapes)
+        output_shapes=self._shapes), self._dtypes, self._shapes)

tensorflow_io/hdf5/BUILD

@@ -10,7 +10,7 @@ load(
 cc_library(
     name = "hdf5_ops",
     srcs = [
-        "kernels/hdf5_input.cc",
+        "kernels/hdf5_kernels.cc",
         "ops/hdf5_ops.cc",
     ],
     copts = tf_io_copts(),

tensorflow_io/hdf5/__init__.py

@@ -15,18 +15,24 @@
 """HDF5 Dataset.
 
 @@HDF5Dataset
+@@list_hdf5_datasets
+@@read_hdf5
 """
 
 from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
 from tensorflow_io.hdf5.python.ops.hdf5_ops import HDF5Dataset
+from tensorflow_io.hdf5.python.ops.hdf5_ops import list_hdf5_datasets
+from tensorflow_io.hdf5.python.ops.hdf5_ops import read_hdf5
 
 from tensorflow.python.util.all_util import remove_undocumented
 
 _allowed_symbols = [
     "HDF5Dataset",
+    "list_hdf5_datasets",
+    "read_hdf5",
 ]
 
 remove_undocumented(__name__, allowed_exception_list=_allowed_symbols)