diff --git a/tensorflow/lite/core/api/flatbuffer_conversions.cc b/tensorflow/lite/core/api/flatbuffer_conversions.cc index 91b3ba65a7e..882b839049c 100644 --- a/tensorflow/lite/core/api/flatbuffer_conversions.cc +++ b/tensorflow/lite/core/api/flatbuffer_conversions.cc @@ -470,6 +470,10 @@ TfLiteStatus ParseOpDataTfLite(const Operator* op, BuiltinOperator op_type, builtin_data); } + case BuiltinOperator_REVERSE_V2: { + return ParseReverseV2(op, error_reporter, allocator, builtin_data); + } + case BuiltinOperator_ROUND: { return ParseRound(op, error_reporter, allocator, builtin_data); } @@ -1006,7 +1010,6 @@ TfLiteStatus ParseOpDataTfLite(const Operator* op, BuiltinOperator op_type, case BuiltinOperator_REAL: case BuiltinOperator_RFFT2D: case BuiltinOperator_SEGMENT_SUM: - case BuiltinOperator_REVERSE_V2: case BuiltinOperator_UNSORTED_SEGMENT_MAX: case BuiltinOperator_UNSORTED_SEGMENT_MIN: case BuiltinOperator_UNSORTED_SEGMENT_PROD: @@ -2467,6 +2470,14 @@ TfLiteStatus ParseStablehloCase(const Operator* op, return kTfLiteError; } +// We have this parse function instead of directly returning kTfLiteOk from the +// switch-case in ParseOpData because this function is used as part of the +// selective registration for the OpResolver implementation in micro. +TfLiteStatus ParseReverseV2(const Operator*, ErrorReporter*, + BuiltinDataAllocator*, void**) { + return kTfLiteOk; +} + // We have this parse function instead of directly returning kTfLiteOk from the // switch-case in ParseOpData because this function is used as part of the // selective registration for the OpResolver implementation in micro. diff --git a/tensorflow/lite/core/api/flatbuffer_conversions.h b/tensorflow/lite/core/api/flatbuffer_conversions.h index 5bc70cc6dea..28aacff6182 100644 --- a/tensorflow/lite/core/api/flatbuffer_conversions.h +++ b/tensorflow/lite/core/api/flatbuffer_conversions.h @@ -318,6 +318,10 @@ TfLiteStatus ParseResizeNearestNeighbor(const Operator* op, BuiltinDataAllocator* allocator, void** builtin_data); +TfLiteStatus ParseReverseV2(const Operator* op, ErrorReporter* error_reporter, + BuiltinDataAllocator* allocator, + void** builtin_data); + TfLiteStatus ParseRound(const Operator* op, ErrorReporter* error_reporter, BuiltinDataAllocator* allocator, void** builtin_data);