Commit d3458112 authored by Rachel Lim's avatar Rachel Lim Committed by TensorFlower Gardener
Browse files

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
parent 885cd294
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment