Skip to content

Automated sync from github.com/tensorflow/tensorflow #2161

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Aug 10, 2023
Merged

Automated sync from github.com/tensorflow/tensorflow #2161

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
4 changes: 4 additions & 0 deletions tensorflow/lite/builtin_ops.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -190,6 +190,10 @@ typedef enum {
kTfLiteBuiltinBitwiseXor = 160,
kTfLiteBuiltinRightShift = 161,
kTfLiteBuiltinStablehloLogistic = 162,
kTfLiteBuiltinStablehloAdd = 163,
kTfLiteBuiltinStablehloDivide = 164,
kTfLiteBuiltinStablehloMultiply = 165,
kTfLiteBuiltinStablehloMaximum = 166,
} TfLiteBuiltinOperator;

#ifdef __cplusplus
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/core/api/flatbuffer_conversions.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/python/schema_py_generated.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
4 changes: 4 additions & 0 deletions tensorflow/lite/schema/schema.fbs
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down
22 changes: 17 additions & 5 deletions tensorflow/lite/schema/schema_generated.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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<size_t>(e);
return EnumNamesBuiltinOperator()[index];
}
Expand Down