Merge pull request #51058 from geetachavan1/cherrypicks_S0AB0

Ensure non-empty rt_nested_splits in tf.raw_ops.RaggedTensorToVariant
tensorflow · Aug 2, 2021 · ecdb9de · ecdb9de
2 parents d80d38e + 801ecc9
commit ecdb9de
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tensorflow/core/kernels/ragged_tensor_to_variant_op.cc b/tensorflow/core/kernels/ragged_tensor_to_variant_op.cc
@@ -157,6 +157,12 @@ class RaggedTensorToVariantOp : public OpKernel {
       return;
     }
 
+    // Checked here instead of at input in case batched_input_ is false
+    OP_REQUIRES(context, ragged_nested_splits_len > 0,
+                errors::InvalidArgument(
+                    "rt_nested_splits must be a list of one or more, but "
+                    "received rt_nested_splits of length 0."));
+
     // Unbatch the Ragged Tensor and encode the components.
     std::vector<RaggedTensorVariant> unbatched_ragged_input;
     auto batched_splits_top_vec =