diff options
-rw-r--r-- | include/tosa_generated.h | 13 | ||||
-rw-r--r-- | python/tosa/Op.py | 1 | ||||
-rw-r--r-- | schema/tosa.fbs | 1 |
3 files changed, 10 insertions, 5 deletions
diff --git a/include/tosa_generated.h b/include/tosa_generated.h index a215891..994b72c 100644 --- a/include/tosa_generated.h +++ b/include/tosa_generated.h @@ -277,11 +277,12 @@ enum Op : uint32_t { Op_DIV_SHAPE = 78, Op_COS = 79, Op_SIN = 80, + Op_CAST_STOCHASTIC = 81, Op_MIN = Op_UNKNOWN, - Op_MAX = Op_SIN + Op_MAX = Op_CAST_STOCHASTIC }; -inline const Op (&EnumValuesOp())[81] { +inline const Op (&EnumValuesOp())[82] { static const Op values[] = { Op_UNKNOWN, Op_ARGMAX, @@ -363,13 +364,14 @@ inline const Op (&EnumValuesOp())[81] { Op_MUL_SHAPE, Op_DIV_SHAPE, Op_COS, - Op_SIN + Op_SIN, + Op_CAST_STOCHASTIC }; return values; } inline const char * const *EnumNamesOp() { - static const char * const names[82] = { + static const char * const names[83] = { "UNKNOWN", "ARGMAX", "AVG_POOL2D", @@ -451,13 +453,14 @@ inline const char * const *EnumNamesOp() { "DIV_SHAPE", "COS", "SIN", + "CAST_STOCHASTIC", nullptr }; return names; } inline const char *EnumNameOp(Op e) { - if (::flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_SIN)) return ""; + if (::flatbuffers::IsOutRange(e, Op_UNKNOWN, Op_CAST_STOCHASTIC)) return ""; const size_t index = static_cast<size_t>(e); return EnumNamesOp()[index]; } diff --git a/python/tosa/Op.py b/python/tosa/Op.py index 35b2b80..021d528 100644 --- a/python/tosa/Op.py +++ b/python/tosa/Op.py @@ -84,3 +84,4 @@ class Op(object): DIV_SHAPE = 78 COS = 79 SIN = 80 + CAST_STOCHASTIC = 81 diff --git a/schema/tosa.fbs b/schema/tosa.fbs index 1a2d952..0686aa7 100644 --- a/schema/tosa.fbs +++ b/schema/tosa.fbs @@ -129,6 +129,7 @@ enum Op:uint32 { DIV_SHAPE, COS, SIN, + CAST_STOCHASTIC, } union Attribute { |