Unverified Commit 3929ffac authored by Mihai Maruseac's avatar Mihai Maruseac Committed by GitHub
Browse files

Merge pull request #50094 from tensorflow/geetachavan1-patch-4

Update ragged_tensor_to_variant_op.cc
No related merge requests found
Showing with 1 addition and 1 deletion
+1 -1
......@@ -176,7 +176,7 @@ class RaggedTensorToVariantOp : public OpKernel {
// Unbatch the Ragged Tensor and encode the components.
std::vector<RaggedTensor> ragged_components;
auto batched_splits_top_vec =
batched_ragged_input.splits(0).vec<SPLIT_TYPE>();
batched_ragged_input.nested_splits[0].vec<SPLIT_TYPE>();
int num_components = batched_splits_top_vec.size() - 1;
OP_REQUIRES(context, num_components >= 0,
errors::Internal("Invalid split argument."));
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment