diff options
Diffstat (limited to 'tensorflow/core/kernels/decode_csv_op.cc')
| -rw-r--r-- | tensorflow/core/kernels/decode_csv_op.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/kernels/decode_csv_op.cc b/tensorflow/core/kernels/decode_csv_op.cc index 3eed847c16..6bfb5bd5bc 100644 --- a/tensorflow/core/kernels/decode_csv_op.cc +++ b/tensorflow/core/kernels/decode_csv_op.cc @@ -61,6 +61,9 @@ class DecodeCSVOp : public OpKernel { OP_REQUIRES_OK(ctx, ctx->input_list("record_defaults", &record_defaults)); for (int i = 0; i < record_defaults.size(); ++i) { + OP_REQUIRES(ctx, record_defaults[i].dims() <= 1, + errors::InvalidArgument( + "Each record default should be at most rank 1")); OP_REQUIRES(ctx, record_defaults[i].NumElements() < 2, errors::InvalidArgument( "There should only be 1 default per field but field ", i, |