Move GetInt8DataPtr function to kernel_util.h

PiperOrigin-RevId: 243924570

tensorflow/lite/kernels/BUILD

@@ -281,6 +281,7 @@ cc_library(
     deps = [
         ":op_macros",
         "//tensorflow/lite/c:c_api_internal",
+        "//tensorflow/lite/kernels:kernel_util",
         "//tensorflow/lite/kernels/internal:kernel_utils",
         "//tensorflow/lite/kernels/internal:tensor_utils",
     ],

tensorflow/lite/kernels/basic_rnn.cc

@@ -27,16 +27,6 @@ namespace ops {
 namespace builtin {
 namespace rnn {
 
-namespace {
-int8_t* GetInt8DataPtr(const TfLiteTensor* tensor, const bool is_uint8) {
-  if (is_uint8) {
-    return reinterpret_cast<int8_t*>(tensor->data.uint8);
-  } else {
-    return tensor->data.int8;
-  }
-}
-}  // namespace
-
 constexpr int kInputTensor = 0;
 constexpr int kWeightsTensor = 1;
 constexpr int kRecurrentWeightsTensor = 2;

tensorflow/lite/kernels/bidirectional_sequence_rnn.cc

@@ -31,18 +31,6 @@ namespace ops {
 namespace builtin {
 namespace bidirectional_sequence_rnn {
 
-namespace {
-
-int8_t* GetInt8DataPtr(const TfLiteTensor* tensor, const bool is_uint8) {
-  if (is_uint8) {
-    return reinterpret_cast<int8_t*>(tensor->data.uint8);
-  } else {
-    return tensor->data.int8;
-  }
-}
-
-}  // namespace
-
 // LINT.IfChange
 
 constexpr int kInputTensor = 0;

tensorflow/lite/kernels/kernel_util.h

@@ -65,6 +65,14 @@ inline const TfLiteTensor* GetOptionalInputTensor(TfLiteContext* context,
   return nullptr;
 }
 
+inline int8_t* GetInt8DataPtr(const TfLiteTensor* tensor, const bool is_uint8) {
+  if (is_uint8) {
+    return reinterpret_cast<int8_t*>(tensor->data.uint8);
+  } else {
+    return tensor->data.int8;
+  }
+}
+
 // Determines whether tensor is constant.
 inline bool IsConstantTensor(const TfLiteTensor* tensor) {
   return tensor->allocation_type == kTfLiteMmapRo;

tensorflow/lite/kernels/lstm_eval.cc

@@ -19,6 +19,7 @@ limitations under the License.
 #include "tensorflow/lite/c/c_api_internal.h"
 #include "tensorflow/lite/kernels/internal/kernel_utils.h"
 #include "tensorflow/lite/kernels/internal/tensor_utils.h"
+#include "tensorflow/lite/kernels/kernel_util.h"
 #include "tensorflow/lite/kernels/op_macros.h"
 
 namespace tflite {
@@ -857,14 +858,6 @@ inline void LstmStepWithAuxInput(
   }
 }
 
-int8_t* GetInt8DataPtr(const TfLiteTensor* tensor, const bool is_uint8) {
-  if (is_uint8) {
-    return reinterpret_cast<int8_t*>(tensor->data.uint8);
-  } else {
-    return tensor->data.int8;
-  }
-}
-
 }  // namespace
 
 TfLiteStatus EvalFloat(