diff options
| author |  Jiri Simsa <jsimsa@google.com> | 2018-07-03 14:02:37 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-03 14:10:20 -0700 |
| commit | a6471888cc9dfe9c18d121149bc0516a3f423fbb (patch) | |
| tree | 75e2d54801ce65e23f478ba86cbb0e51451aa7bb /tensorflow/core/kernels/data/window_dataset.cc | |
| parent | b6ebb965937a62e15299cec7c91896fd011ce416 (diff) | |
[tf.data] Experimental transformations for windowing and batching of windows.
This change introduces the `window` tf.data transformation, which can be used to create windows of elements (represented as a dataset) from a dataset. This transformation enables applying different batching logic to different components of a dataset.
To illustrate the benefits of the transformation, this CL also introduces transformations for batching and padded batching of windows of both dense and sparse tensors. Notably, padded batching of sparse tensors was previously not possible.
PiperOrigin-RevId: 203179522
Diffstat (limited to 'tensorflow/core/kernels/data/window_dataset.cc')
| -rw-r--r-- | tensorflow/core/kernels/data/window_dataset.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/data/window_dataset.cc b/tensorflow/core/kernels/data/window_dataset.cc index 668b461374..17551bccd9 100644 --- a/tensorflow/core/kernels/data/window_dataset.cc +++ b/tensorflow/core/kernels/data/window_dataset.cc @@ -17,6 +17,7 @@ limitations under the License. namespace tensorflow { namespace { +// TODO(b/110981596): Support checkpointing. class WindowDataset : public DatasetBase { public: WindowDataset(std::vector<std::vector<Tensor>> elements, |