27 lines
1.2 KiB
Diff
27 lines
1.2 KiB
Diff
From 2e0ee46f1a47675152d3d865797a18358881d7a6 Mon Sep 17 00:00:00 2001
|
|
From: Laura Pak <lpak@google.com>
|
|
Date: Tue, 27 Jul 2021 12:30:33 -0700
|
|
Subject: [PATCH] Ensure non-empty input_splits in tf.raw_ops.UnicodeEncode
|
|
|
|
PiperOrigin-RevId: 387170080
|
|
Change-Id: I3b489acc51c5cb4124c535b9df7cc6e62ef21766
|
|
---
|
|
tensorflow/core/kernels/unicode_ops.cc | 4 ++++
|
|
1 file changed, 4 insertions(+)
|
|
|
|
diff --git a/tensorflow/core/kernels/unicode_ops.cc b/tensorflow/core/kernels/unicode_ops.cc
|
|
index 301cdf932ba60..d4153e04e4f94 100644
|
|
--- a/tensorflow/core/kernels/unicode_ops.cc
|
|
+++ b/tensorflow/core/kernels/unicode_ops.cc
|
|
@@ -533,6 +533,10 @@ class UnicodeEncodeOp : public OpKernel {
|
|
const Tensor& input_splits = context->input(1);
|
|
const auto input_splits_flat = input_splits.flat<SPLITS_TYPE>();
|
|
|
|
+ OP_REQUIRES(
|
|
+ context, input_splits.NumElements() > 0,
|
|
+ errors::InvalidArgument("Input_splits should contain elements, but "
|
|
+ "given input_values has 0 elements"));
|
|
// Operation will treat first argument in input_splits as if it were zero
|
|
// regardless of its actual value since splits should begin with zero and
|
|
// end with the length of the input values vector.
|