From 3acb1cbfdd492ab4f657799ed0c2279a5e390248 Mon Sep 17 00:00:00 2001 From: Won Jeon Date: Wed, 7 Jun 2023 23:26:32 +0000 Subject: Add ERF operator to serialization_lib Signed-off-by: Won Jeon Change-Id: Idc253e51e7a613cfc0c7da653073a71bf0fcb162 --- include/tosa_generated.h | 13 ++++++++----- python/tosa/Op.py | 1 + schema/tosa.fbs | 1 + 3 files changed, 10 insertions(+), 5 deletions(-) diff --git a/include/tosa_generated.h b/include/tosa_generated.h index 5549614..422ae18 100644 --- a/include/tosa_generated.h +++ b/include/tosa_generated.h @@ -257,11 +257,12 @@ enum Op : uint32_t { Op_WHILE_LOOP = 68, Op_FFT2D = 69, Op_RFFT2D = 70, + Op_ERF = 71, Op_MIN = Op_UNKNOWN, - Op_MAX = Op_RFFT2D + Op_MAX = Op_ERF }; -inline const Op (&EnumValuesOp())[71] { +inline const Op (&EnumValuesOp())[72] { static const Op values[] = { Op_UNKNOWN, Op_ARGMAX, @@ -333,13 +334,14 @@ inline const Op (&EnumValuesOp())[71] { Op_COND_IF, Op_WHILE_LOOP, Op_FFT2D, - Op_RFFT2D + Op_RFFT2D, + Op_ERF }; return values; } inline const char * const *EnumNamesOp() { - static const char * const names[72] = { + static const char * const names[73] = { "UNKNOWN", "ARGMAX", "AVG_POOL2D", @@ -411,13 +413,14 @@ inline const char * const *EnumNamesOp() { "WHILE_LOOP", "FFT2D", "RFFT2D", + "ERF", nullptr }; return names; } inline const char *EnumNameOp(Op e) { - if (flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_RFFT2D)) return ""; + if (flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_ERF)) return ""; const size_t index = static_cast(e); return EnumNamesOp()[index]; } diff --git a/python/tosa/Op.py b/python/tosa/Op.py index b8e68c0..b310311 100644 --- a/python/tosa/Op.py +++ b/python/tosa/Op.py @@ -74,3 +74,4 @@ class Op(object): WHILE_LOOP = 68 FFT2D = 69 RFFT2D = 70 + ERF = 71 diff --git a/schema/tosa.fbs b/schema/tosa.fbs index eeed385..057933c 100644 --- a/schema/tosa.fbs +++ b/schema/tosa.fbs @@ -116,6 +116,7 @@ enum Op:uint32 { WHILE_LOOP, FFT2D, RFFT2D, + ERF, } union Attribute { -- cgit v1.2.1