diff --git a/tensorflow/lite/builtin_ops.h b/tensorflow/lite/builtin_ops.h index 68459bb36ed..a3a77be512c 100644 --- a/tensorflow/lite/builtin_ops.h +++ b/tensorflow/lite/builtin_ops.h @@ -190,6 +190,10 @@ typedef enum { kTfLiteBuiltinBitwiseXor = 160, kTfLiteBuiltinRightShift = 161, kTfLiteBuiltinStablehloLogistic = 162, + kTfLiteBuiltinStablehloAdd = 163, + kTfLiteBuiltinStablehloDivide = 164, + kTfLiteBuiltinStablehloMultiply = 165, + kTfLiteBuiltinStablehloMaximum = 166, } TfLiteBuiltinOperator; #ifdef __cplusplus diff --git a/tensorflow/lite/core/api/flatbuffer_conversions.cc b/tensorflow/lite/core/api/flatbuffer_conversions.cc index 0f6f7fddc7f..3afe6829692 100644 --- a/tensorflow/lite/core/api/flatbuffer_conversions.cc +++ b/tensorflow/lite/core/api/flatbuffer_conversions.cc @@ -900,6 +900,10 @@ TfLiteStatus ParseOpDataTfLite(const Operator* op, BuiltinOperator op_type, case BuiltinOperator_BITCAST: case BuiltinOperator_WHERE: case BuiltinOperator_STABLEHLO_LOGISTIC: + case BuiltinOperator_STABLEHLO_ADD: + case BuiltinOperator_STABLEHLO_DIVIDE: + case BuiltinOperator_STABLEHLO_MULTIPLY: + case BuiltinOperator_STABLEHLO_MAXIMUM: return kTfLiteOk; case BuiltinOperator_PLACEHOLDER_FOR_GREATER_OP_CODES: return kTfLiteError; diff --git a/tensorflow/lite/python/schema_py_generated.py b/tensorflow/lite/python/schema_py_generated.py index df30a696c36..80caf3181ca 100755 --- a/tensorflow/lite/python/schema_py_generated.py +++ b/tensorflow/lite/python/schema_py_generated.py @@ -1552,6 +1552,10 @@ class BuiltinOperator(object): BITWISE_XOR = 160 RIGHT_SHIFT = 161 STABLEHLO_LOGISTIC = 162 + STABLEHLO_ADD = 163 + STABLEHLO_DIVIDE = 164 + STABLEHLO_MULTIPLY = 165 + STABLEHLO_MAXIMUM = 166 # automatically generated by the FlatBuffers compiler, do not modify # namespace: tflite diff --git a/tensorflow/lite/schema/schema.fbs b/tensorflow/lite/schema/schema.fbs index 9120e345d3a..8c1a37af65a 100644 --- a/tensorflow/lite/schema/schema.fbs +++ b/tensorflow/lite/schema/schema.fbs @@ -423,6 +423,10 @@ enum BuiltinOperator : int32 { // All Operators start with STABLEHLO_ prefixes are subject to change // Many of the ops below can not be executed by TFlite runtime STABLEHLO_LOGISTIC = 162, // WARNING: Do not have runtime support + STABLEHLO_ADD = 163, // WARNING: No runtime support yet + STABLEHLO_DIVIDE = 164, // WARNING: No runtime support yet + STABLEHLO_MULTIPLY = 165, // WARNING: No runtime support yet + STABLEHLO_MAXIMUM = 166, // WARNING: No runtime support yet } // LINT.ThenChange(nnapi_linter/linter.proto) diff --git a/tensorflow/lite/schema/schema_generated.h b/tensorflow/lite/schema/schema_generated.h index 4903b48fe10..7851aa3fd7d 100755 --- a/tensorflow/lite/schema/schema_generated.h +++ b/tensorflow/lite/schema/schema_generated.h @@ -1075,11 +1075,15 @@ enum BuiltinOperator : int32_t { BuiltinOperator_BITWISE_XOR = 160, BuiltinOperator_RIGHT_SHIFT = 161, BuiltinOperator_STABLEHLO_LOGISTIC = 162, + BuiltinOperator_STABLEHLO_ADD = 163, + BuiltinOperator_STABLEHLO_DIVIDE = 164, + BuiltinOperator_STABLEHLO_MULTIPLY = 165, + BuiltinOperator_STABLEHLO_MAXIMUM = 166, BuiltinOperator_MIN = BuiltinOperator_ADD, - BuiltinOperator_MAX = BuiltinOperator_STABLEHLO_LOGISTIC + BuiltinOperator_MAX = BuiltinOperator_STABLEHLO_MAXIMUM }; -inline const BuiltinOperator (&EnumValuesBuiltinOperator())[163] { +inline const BuiltinOperator (&EnumValuesBuiltinOperator())[167] { static const BuiltinOperator values[] = { BuiltinOperator_ADD, BuiltinOperator_AVERAGE_POOL_2D, @@ -1243,13 +1247,17 @@ inline const BuiltinOperator (&EnumValuesBuiltinOperator())[163] { BuiltinOperator_BITCAST, BuiltinOperator_BITWISE_XOR, BuiltinOperator_RIGHT_SHIFT, - BuiltinOperator_STABLEHLO_LOGISTIC + BuiltinOperator_STABLEHLO_LOGISTIC, + BuiltinOperator_STABLEHLO_ADD, + BuiltinOperator_STABLEHLO_DIVIDE, + BuiltinOperator_STABLEHLO_MULTIPLY, + BuiltinOperator_STABLEHLO_MAXIMUM }; return values; } inline const char * const *EnumNamesBuiltinOperator() { - static const char * const names[164] = { + static const char * const names[168] = { "ADD", "AVERAGE_POOL_2D", "CONCATENATION", @@ -1413,13 +1421,17 @@ inline const char * const *EnumNamesBuiltinOperator() { "BITWISE_XOR", "RIGHT_SHIFT", "STABLEHLO_LOGISTIC", + "STABLEHLO_ADD", + "STABLEHLO_DIVIDE", + "STABLEHLO_MULTIPLY", + "STABLEHLO_MAXIMUM", nullptr }; return names; } inline const char *EnumNameBuiltinOperator(BuiltinOperator e) { - if (flatbuffers::IsOutRange(e, BuiltinOperator_ADD, BuiltinOperator_STABLEHLO_LOGISTIC)) return ""; + if (flatbuffers::IsOutRange(e, BuiltinOperator_ADD, BuiltinOperator_STABLEHLO_MAXIMUM)) return ""; const size_t index = static_cast(e); return EnumNamesBuiltinOperator()[index]; }