diff --git a/tensorflow/contrib/data/python/ops/readers.py b/tensorflow/contrib/data/python/ops/readers.py index 2e759f09a1f..1ea118607d2 100644 --- a/tensorflow/contrib/data/python/ops/readers.py +++ b/tensorflow/contrib/data/python/ops/readers.py @@ -17,6 +17,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +from tensorflow.python.compat import compat from tensorflow.python.data.experimental.ops import readers from tensorflow.python.data.ops import dataset_ops from tensorflow.python.data.ops import readers as core_readers @@ -390,8 +391,12 @@ class LMDBDataset(dataset_ops.DatasetSource): """ self._filenames = ops.convert_to_tensor( filenames, dtype=dtypes.string, name="filenames") - variant_tensor = gen_experimental_dataset_ops.experimental_lmdb_dataset( - self._filenames, **self._flat_structure) + if compat.forward_compatible(2019, 8, 3): + variant_tensor = gen_experimental_dataset_ops.lmdb_dataset( + self._filenames, **self._flat_structure) + else: + variant_tensor = gen_experimental_dataset_ops.experimental_lmdb_dataset( + self._filenames, **self._flat_structure) super(LMDBDataset, self).__init__(variant_tensor) @property diff --git a/tensorflow/contrib/data/python/ops/sliding.py b/tensorflow/contrib/data/python/ops/sliding.py index 672ead9c117..c329b6559f9 100644 --- a/tensorflow/contrib/data/python/ops/sliding.py +++ b/tensorflow/contrib/data/python/ops/sliding.py @@ -17,6 +17,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function +from tensorflow.python.compat import compat from tensorflow.python.data.ops import dataset_ops from tensorflow.python.data.util import nest from tensorflow.python.framework import dtypes @@ -41,12 +42,20 @@ class _SlideDataset(dataset_ops.UnaryDataset): input_structure = dataset_ops.get_structure(input_dataset) self._element_spec = nest.map_structure( lambda component_spec: component_spec._batch(None), input_structure) # pylint: disable=protected-access - variant_tensor = ged_ops.experimental_sliding_window_dataset( - self._input_dataset._variant_tensor, # pylint: disable=protected-access - window_size=self._window_size, - window_shift=self._window_shift, - window_stride=self._window_stride, - **self._flat_structure) + if compat.forward_compatible(2019, 8, 3): + variant_tensor = ged_ops.sliding_window_dataset( + self._input_dataset._variant_tensor, # pylint: disable=protected-access + window_size=self._window_size, + window_shift=self._window_shift, + window_stride=self._window_stride, + **self._flat_structure) + else: + variant_tensor = ged_ops.experimental_sliding_window_dataset( + self._input_dataset._variant_tensor, # pylint: disable=protected-access + window_size=self._window_size, + window_shift=self._window_shift, + window_stride=self._window_stride, + **self._flat_structure) super(_SlideDataset, self).__init__(input_dataset, variant_tensor) @property diff --git a/tensorflow/core/api_def/base_api/api_def_AssertNextDataset.pbtxt b/tensorflow/core/api_def/base_api/api_def_AssertNextDataset.pbtxt new file mode 100644 index 00000000000..c97b807713f --- /dev/null +++ b/tensorflow/core/api_def/base_api/api_def_AssertNextDataset.pbtxt @@ -0,0 +1,4 @@ +op { + graph_op_name: "AssertNextDataset" + visibility: HIDDEN +} diff --git a/tensorflow/core/api_def/base_api/api_def_AutoShardDataset.pbtxt b/tensorflow/core/api_def/base_api/api_def_AutoShardDataset.pbtxt new file mode 100644 index 00000000000..403499e9b6f --- /dev/null +++ b/tensorflow/core/api_def/base_api/api_def_AutoShardDataset.pbtxt @@ -0,0 +1,32 @@ +op { + graph_op_name: "AutoShardDataset" + visibility: HIDDEN + in_arg { + name: "input_dataset" + description: <