diff options
author | Rachel Lim <rachelim@google.com> | 2018-09-13 14:18:16 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-13 14:26:13 -0700 |
commit | d3458112ad5a1612ec6c77f7de4a0e0ec801e882 (patch) | |
tree | d1b265fed67f2f4a9b502ea6eb32fe7c26eab2ee /tensorflow/core/api_def | |
parent | 885cd2942ae7b6239146a3f51ec3d6948ac2b89e (diff) |
Consistency in record_default shapes for tf.contrib.data.CsvDataset & tf.decode_csv:
- Modify shape assertions so that both graph and eager accept rank 0 (scalar) and rank 1 tensors as `record_defaults`, and raise an error on other shapes.
- Make tests run in both graph and eager modes
Fixes #22030.
PiperOrigin-RevId: 212877058
Diffstat (limited to 'tensorflow/core/api_def')
-rw-r--r-- | tensorflow/core/api_def/base_api/api_def_DecodeCSV.pbtxt | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/tensorflow/core/api_def/base_api/api_def_DecodeCSV.pbtxt b/tensorflow/core/api_def/base_api/api_def_DecodeCSV.pbtxt index e39213cbc7..440800704e 100644 --- a/tensorflow/core/api_def/base_api/api_def_DecodeCSV.pbtxt +++ b/tensorflow/core/api_def/base_api/api_def_DecodeCSV.pbtxt @@ -11,7 +11,8 @@ END name: "record_defaults" description: <<END One tensor per column of the input record, with either a -scalar default value for that column or empty if the column is required. +scalar default value for that column or an empty vector if the column is +required. END } out_arg { |