From b90d7302bd1d3c028e894cd823428e707478ca1f Mon Sep 17 00:00:00 2001 From: "TF.Text Team" Date: Wed, 8 Jan 2025 04:21:19 -0800 Subject: [PATCH] No public description PiperOrigin-RevId: 713241564 --- .../core/kernels/ragged_tensor_to_tensor_tflite_test.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc b/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc index 4de599821..f93d1efd2 100644 --- a/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc +++ b/tensorflow_text/core/kernels/ragged_tensor_to_tensor_tflite_test.cc @@ -100,7 +100,9 @@ class RaggedTensorToTensorOpModel : public SingleOpModel { std::vector GetOutputShape() { return GetTensorShape(output_); } std::vector GetOutputFloat() { return ExtractVector(output_); } - std::vector GetOutputInt() { return ExtractVector(output_); } + std::vector GetOutputInt() { + return ExtractVector(output_); + } void InvokeFloat(const std::vector& shape, const std::vector& values, float default_value, @@ -114,7 +116,7 @@ class RaggedTensorToTensorOpModel : public SingleOpModel { SingleOpModel::Invoke(); } void InvokeInt(const std::vector& shape, - const std::vector& values, int32 default_value, + const std::vector& values, int32_t default_value, const std::vector>& partition_values) { PopulateTensor(input_shape_, shape); PopulateTensor(input_values_, values);