StreamingFilesDataset fixes (#19413)
authorNand Dalal <nand@clarifai.com>
Mon, 21 May 2018 03:15:21 +0000 (22:15 -0500)
committerShanqing Cai <cais@google.com>
Mon, 21 May 2018 03:15:21 +0000 (23:15 -0400)
commita2e1b4dcbd0ecd310efa2eb258dcbdbcf942af86
treeaf2bfccb29da1cf4f72bc92569dbe71269501808
parente844159e63af0c4654dba98657e19f590a4edbb8
StreamingFilesDataset fixes (#19413)

* use source_dataset.output_dtypes to yield correctly typed output dataset

* add test and fix issue introduced by 2a6c5998a239f41926ca295ac20bb595862fd5ff
tensorflow/contrib/tpu/python/tpu/datasets.py
tensorflow/contrib/tpu/python/tpu/datasets_test.py