Sync from upstream TF.

Author: TFLM-bot

tensorflow/lite/core/api/flatbuffer_conversions.cc

@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "tensorflow/lite/core/api/flatbuffer_conversions.h"
 
+#include <algorithm>
 #include <cstddef>
 #include <cstdint>
 #include <memory>
@@ -881,6 +882,10 @@ TfLiteStatus ParseOpDataTfLite(const Operator* op, BuiltinOperator op_type,
     case BuiltinOperator_STABLEHLO_GATHER: {
       return ParseStablehloGather(op, error_reporter, allocator, builtin_data);
     }
+    case BuiltinOperator_STABLEHLO_REDUCE_WINDOW: {
+      return ParseStablehloReduceWindow(op, error_reporter, allocator,
+                                        builtin_data);
+    }
     case BuiltinOperator_REDUCE_WINDOW: {
       auto params = safe_allocator.Allocate<TfLiteReduceWindowParams>();
       TF_LITE_ENSURE(error_reporter, params != nullptr);
@@ -949,7 +954,6 @@ TfLiteStatus ParseOpDataTfLite(const Operator* op, BuiltinOperator op_type,
     case BuiltinOperator_STABLEHLO_CONVERT:
     case BuiltinOperator_STABLEHLO_PAD:
     case BuiltinOperator_STABLEHLO_DOT_GENERAL:
-    case BuiltinOperator_STABLEHLO_REDUCE_WINDOW:
     case BuiltinOperator_STABLEHLO_SORT:
     case BuiltinOperator_STABLEHLO_WHILE:
     case BuiltinOperator_STABLEHLO_TRANSPOSE:
@@ -2096,6 +2100,98 @@ TfLiteStatus ParseResizeNearestNeighbor(const Operator* op,
   return kTfLiteOk;
 }
 
+TfLiteStatus ParseStablehloReduceWindow(const Operator* op,
+                                        ErrorReporter* error_reporter,
+                                        BuiltinDataAllocator* allocator,
+                                        void** builtin_data) {
+  CheckParsePointerParams(op, error_reporter, allocator, builtin_data);
+
+  SafeBuiltinDataAllocator safe_allocator(allocator);
+  auto params = safe_allocator.Allocate<TfLiteStablehloReduceWindowParams>();
+
+  const StablehloReduceWindowOptions* schema_params =
+      op->builtin_options_2_as_StablehloReduceWindowOptions();
+  if (schema_params) {
+    if (!schema_params->window_dimensions() ||
+        schema_params->window_dimensions()->size() == 0) {
+      TF_LITE_REPORT_ERROR(error_reporter,
+                           "'window_dimensions' attribute is not optional for "
+                           "'stablehlo.reduce_window' and cannot be empty.");
+      return kTfLiteError;
+    }
+
+    const size_t rank = schema_params->window_dimensions()->size();
+
+    auto LoadAttr = [&error_reporter](
+                        auto& params_array, auto* const flatbuffer_vector,
+                        const char* attr_name, const size_t expected_size,
+                        const int64_t fill_value) -> TfLiteStatus {
+      if (flatbuffer_vector && flatbuffer_vector->size()) {
+        if (expected_size != 0 && flatbuffer_vector->size() != expected_size) {
+          TF_LITE_REPORT_ERROR(
+              error_reporter,
+              "'%s' attribute of 'stablehlo.reduce_window' does not have the "
+              "expected size (%llu != %llu).",
+              attr_name, flatbuffer_vector->size(), expected_size);
+          return kTfLiteError;
+        }
+        TfLiteStatus status = FlatBufferIntVectorToArray(
+            sizeof(params_array), flatbuffer_vector, params_array,
+            error_reporter, "stablehlo.reduce_window");
+        if (status != kTfLiteOk) {
+          TF_LITE_REPORT_ERROR(error_reporter, "Check the '%s' attribute.",
+                               attr_name);
+          return status;
+        }
+      } else {
+        std::fill_n(params_array,
+                    TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT,
+                    fill_value);
+      }
+      return kTfLiteOk;
+    };
+
+    if (TfLiteStatus status = LoadAttr(
+            params->window_dimensions, schema_params->window_dimensions(),
+            "window_dimensions", /*expected_size=*/rank, /*fill_value=*/1);
+        status != kTfLiteOk) {
+      return status;
+    }
+    if (TfLiteStatus status = LoadAttr(
+            params->window_strides, schema_params->window_strides(),
+            "window_strides", /*expected_size=*/rank, /*fill_value=*/1);
+        status != kTfLiteOk) {
+      return status;
+    }
+    if (TfLiteStatus status = LoadAttr(
+            params->base_dilations, schema_params->base_dilations(),
+            "base_dilations", /*expected_size=*/rank, /*fill_value=*/1);
+        status != kTfLiteOk) {
+      return status;
+    }
+    if (TfLiteStatus status = LoadAttr(
+            params->window_dilations, schema_params->window_dilations(),
+            "window_dilations", /*expected_size=*/rank, /*fill_value=*/1);
+        status != kTfLiteOk) {
+      return status;
+    }
+    if (TfLiteStatus status =
+            LoadAttr(params->padding, schema_params->padding(), "padding",
+                     /*expected_size=*/2 * rank, /*fill_value=*/0);
+        status != kTfLiteOk) {
+      return status;
+    }
+
+    params->body_subgraph_index = schema_params->body_subgraph_index();
+    *builtin_data = params.release();
+    return kTfLiteOk;
+  }
+  TF_LITE_REPORT_ERROR(
+      error_reporter,
+      "Could not get 'stablehlo.reduce_window' operation parameters.");
+  return kTfLiteError;
+}
+
 TfLiteStatus ParseStablehloScatter(const Operator* op,
                                    ErrorReporter* error_reporter,
                                    BuiltinDataAllocator* allocator,

tensorflow/lite/core/api/flatbuffer_conversions.h

@@ -435,6 +435,11 @@ TfLiteStatus ParseStablehloGather(const Operator* op,
                                   BuiltinDataAllocator* allocator,
                                   void** builtin_data);
 
+TfLiteStatus ParseStablehloReduceWindow(const Operator* op,
+                                        ErrorReporter* error_reporter,
+                                        BuiltinDataAllocator* allocator,
+                                        void** builtin_data);
+
 }  // namespace tflite
 
 #endif  // TENSORFLOW_LITE_CORE_API_FLATBUFFER_CONVERSIONS_H_

tensorflow/lite/core/c/builtin_op_data.h

@@ -34,6 +34,7 @@ extern "C" {
 #define TFLITE_RESHAPE_PARAMS_MAX_DIMENSION_COUNT 8
 #define TFLITE_STABLEHLO_SCATTER_PARAMS_MAX_DIMENSION_COUNT 8
 #define TFLITE_STABLEHLO_GATHER_PARAMS_MAX_DIMENSION_COUNT 8
+#define TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT 8
 
 // TODO(aselle): Consider using "if this then that" for testing.
 
@@ -605,6 +606,22 @@ typedef struct {
   bool indices_are_sorted;
 } TfLiteStablehloGatherParams;
 
+typedef struct {
+  // See the stablehlo spec for the explanation of the attributes:
+  // https://github.com/openxla/stablehlo/blob/main/docs/spec.md#reduce_window
+  int64_t window_dimensions
+      [TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT];
+  int64_t
+      window_strides[TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT];
+  int64_t
+      base_dilations[TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT];
+  int64_t window_dilations
+      [TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT];
+  int64_t
+      padding[2 * TFLITE_STABLEHLO_REDUCE_WINDOW_PARAMS_MAX_DIMENSION_COUNT];
+  int body_subgraph_index;
+} TfLiteStablehloReduceWindowParams;
+
 enum TfLiteReduceWindowFunction {
   TfLiteReduceWindowFunctionUnsupported,
   TfLiteReduceWindowFunctionAdd,

tensorflow/lite/core/c/c_api_types.h

@@ -34,9 +34,13 @@ limitations under the License.
 extern "C" {
 #endif
 
-/** \addtogroup c_api_types tensorflow/lite/c/c_api_types.h
+// clang-format off
+// NOLINTBEGIN(whitespace/line_length)
+/** \defgroup c_api_types tensorflow/lite/c/c_api_types.h
  *  @{
  */
+// NOLINTEND(whitespace/line_length)
+// clang-format on
 
 // Define TFL_CAPI_EXPORT macro to export a function properly with a shared
 // library.

tensorflow/lite/core/c/common.h

@@ -54,9 +54,13 @@ limitations under the License.
 extern "C" {
 #endif  // __cplusplus
 
-/** \addtogroup common tensorflow/lite/c/common.h
+// clang-format off
+// NOLINTBEGIN(whitespace/line_length)
+/** \defgroup common tensorflow/lite/c/common.h
  *  @{
  */
+// NOLINTEND(whitespace/line_length)
+// clang-format on
 
 /// The list of external context types known to TF Lite. This list exists solely
 /// to avoid conflicts and to ensure ops can share the external contexts they

tensorflow/lite/schema/schema.fbs

@@ -429,7 +429,7 @@ enum BuiltinOperator : int32 {
   STABLEHLO_LOGISTIC = 162, // WARNING: Do not have runtime support
   STABLEHLO_ADD = 163,
   STABLEHLO_DIVIDE = 164, // WARNING: No runtime support yet
-  STABLEHLO_MULTIPLY = 165, // WARNING: No runtime support yet
+  STABLEHLO_MULTIPLY = 165,
   STABLEHLO_MAXIMUM = 166, // WARNING: No runtime support yet
   STABLEHLO_RESHAPE = 167, // WARNING: No runtime support yet
   STABLEHLO_CLAMP = 168, // WARNING: No runtime support
@@ -462,14 +462,14 @@ enum BuiltinOperator : int32 {
   STABLEHLO_PAD = 195, // WARNING: No runtime support
   STABLEHLO_IOTA = 196, // WARNING: No runtime support
   STABLEHLO_DOT_GENERAL = 197, // WARNING: No runtime support
-  STABLEHLO_REDUCE_WINDOW = 198, // WARNING: No runtime support
+  STABLEHLO_REDUCE_WINDOW = 198,
   STABLEHLO_SORT = 199, // WARNING: No runtime support
   STABLEHLO_WHILE = 200, // WARNING: No runtime support
   STABLEHLO_GATHER = 201,
   STABLEHLO_TRANSPOSE = 202, // WARNING: No runtime support
   DILATE = 203,
   STABLEHLO_RNG_BIT_GENERATOR = 204,
-  REDUCE_WINDOW = 205,
+  REDUCE_WINDOW = 205 (deprecated),
 }
 // LINT.ThenChange(nnapi_linter/linter.proto)
 
@@ -626,7 +626,7 @@ union BuiltinOptions2{
   StablehloTransposeOptions,
   DilateOptions,
   StablehloRngBitGeneratorOptions,
-  ReduceWindowOptions,
+  ReduceWindowOptions (deprecated),
 }
 
 table StablehloGatherOptions{
@@ -1458,7 +1458,7 @@ enum ReduceWindowFunction : int {
   ANY,
 }
 
-table ReduceWindowOptions{
+table ReduceWindowOptions (deprecated) {
   reduce_function: ReduceWindowFunction;
 }