Merge pull request #51256 from tensorflow/mm-cherrypick-be7a4de6adfbd…

…303ce08be4332554dff70362612-on-r2.3 Ensure non-empty rt_nested_splits in tf.raw_ops.RaggedTensorToVariant
tensorflow · Aug 5, 2021 · aa7401d · aa7401d
2 parents 6c1eb96 + c3ef7bc
commit aa7401d
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
@@ -173,6 +173,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<RaggedTensor> ragged_components;
     auto batched_splits_top_vec =