experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Change the implementation of getter methods of schema utils.

Browse Source 
This is to address the problem of supporting quantized models that contain
zero-valued deprecated builtin code for Dequantize ops in the newer version of
the runtime.

PiperOrigin-RevId: 336765278
Change-Id: I668698bc01145799041672b2e0e9ebc5858ed979
This commit is contained in:
Jaesung Chung 2020-10-12 16:09:39 -07:00 committed by TensorFlower Gardener
parent 35aeaec772
commit cf28969fb2
1 changed files with 7 additions and 6 deletions

13
tensorflow/lite/schema/schema_utils.cc
View File

@ -14,6 +14,8 @@ limitations under the License.
==============================================================================*/
#include "tensorflow/lite/schema/schema_utils.h"
#include <algorithm>
#include "tensorflow/lite/kernels/internal/compatibility.h"
namespace tflite {
@ -44,18 +46,17 @@ BuiltinOperator GetBuiltinCode(const OperatorCode *op_code) {
// Caller should guarantee that the given argument value is not a nullptr.
TFLITE_DCHECK(op_code != nullptr);
return (op_code->builtin_code() ? op_code->builtin_code()
: static_cast<BuiltinOperator>(
op_code->deprecated_builtin_code()));
return std::max(
op_code->builtin_code(),
static_cast<BuiltinOperator>(op_code->deprecated_builtin_code()));
}
BuiltinOperator GetBuiltinCode(const OperatorCodeT *op_code) {
// Caller should guarantee that the given argument value is not a nullptr.
TFLITE_DCHECK(op_code != nullptr);
return (op_code->builtin_code
? op_code->builtin_code
: static_cast<BuiltinOperator>(op_code->deprecated_builtin_code));
return std::max(op_code->builtin_code, static_cast<BuiltinOperator>(
op_code->deprecated_builtin_code));
}
int8_t ConvertBuiltinCodeToDeprecatedBuiltinCode(