33 lines
1.2 KiB
Diff
33 lines
1.2 KiB
Diff
From be7a4de6adfbd303ce08be4332554dff70362612 Mon Sep 17 00:00:00 2001
|
|
From: Laura Pak <lpak@google.com>
|
|
Date: Thu, 29 Jul 2021 14:05:34 -0700
|
|
Subject: [PATCH] Ensure non-empty rt_nested_splits in
|
|
tf.raw_ops.RaggedTensorToVariant
|
|
|
|
PiperOrigin-RevId: 387664237
|
|
Change-Id: Ia1700c34b5610873d63561abc86e23b46ead93b3
|
|
---
|
|
tensorflow/core/kernels/ragged_tensor_to_variant_op.cc | 6 ++++++
|
|
1 file changed, 6 insertions(+)
|
|
|
|
diff --git a/tensorflow/core/kernels/ragged_tensor_to_variant_op.cc b/tensorflow/core/kernels/ragged_tensor_to_variant_op.cc
|
|
index 7a5ae1c6..3190534b 100644
|
|
--- 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;
|
|
OP_REQUIRES_OK(context, UnbatchRaggedZerothDim<VALUE_TYPE, SPLIT_TYPE>(
|
|
--
|
|
2.27.0
|
|
|