No public description

PiperOrigin-RevId: 713241564
tensorflow · Jan 8, 2025 · b90d730 · b90d730
1 parent 87715fb
commit b90d730
Showing 1 changed file with 4 additions and 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
@@ -100,7 +100,9 @@ class RaggedTensorToTensorOpModel : public SingleOpModel {
   std::vector<int> GetOutputShape() { return GetTensorShape(output_); }
 
   std::vector<float> GetOutputFloat() { return ExtractVector<float>(output_); }
-  std::vector<int32> GetOutputInt() { return ExtractVector<int32>(output_); }
+  std::vector<int32_t> GetOutputInt() {
+    return ExtractVector<int32_t>(output_);
+  }
 
   void InvokeFloat(const std::vector<int>& shape,
                    const std::vector<float>& values, float default_value,
@@ -114,7 +116,7 @@ class RaggedTensorToTensorOpModel : public SingleOpModel {
     SingleOpModel::Invoke();
   }
   void InvokeInt(const std::vector<int>& shape,
-                 const std::vector<int32>& values, int32 default_value,
+                 const std::vector<int32_t>& values, int32_t default_value,
                  const std::vector<std::vector<int>>& partition_values) {
     PopulateTensor(input_shape_, shape);
     PopulateTensor(input_values_, values);