diff options
author | 2018-07-16 19:09:52 -0700 | |
---|---|---|
committer | 2018-07-16 19:13:36 -0700 | |
commit | 1a2af489b1087eb22ec76863867e4e397e453e34 (patch) | |
tree | d2576dbbfc6f77897e59353e0844aa0c75a23033 /tensorflow/contrib/lite/schema | |
parent | d8f3425e5b054dff01b5ece80e8c8a101c4ed816 (diff) |
Support reduce_max and reduce_prod
PiperOrigin-RevId: 204846139
Diffstat (limited to 'tensorflow/contrib/lite/schema')
-rw-r--r-- | tensorflow/contrib/lite/schema/schema.fbs | 4 | ||||
-rwxr-xr-x | tensorflow/contrib/lite/schema/schema_generated.h | 12 |
2 files changed, 12 insertions, 4 deletions
diff --git a/tensorflow/contrib/lite/schema/schema.fbs b/tensorflow/contrib/lite/schema/schema.fbs index 64830b1dc3..6c3189a884 100644 --- a/tensorflow/contrib/lite/schema/schema.fbs +++ b/tensorflow/contrib/lite/schema/schema.fbs @@ -155,13 +155,15 @@ enum BuiltinOperator : byte { EQUAL = 71, NOT_EQUAL = 72, LOG = 73, - SUM=74, + SUM = 74, SQRT = 75, RSQRT = 76, SHAPE = 77, POW = 78, ARG_MIN = 79, FAKE_QUANT = 80, + REDUCE_PROD = 81, + REDUCE_MAX = 82, } // Options for the builtin operators. diff --git a/tensorflow/contrib/lite/schema/schema_generated.h b/tensorflow/contrib/lite/schema/schema_generated.h index c0b57039cb..8052404319 100755 --- a/tensorflow/contrib/lite/schema/schema_generated.h +++ b/tensorflow/contrib/lite/schema/schema_generated.h @@ -351,11 +351,13 @@ enum BuiltinOperator { BuiltinOperator_POW = 78, BuiltinOperator_ARG_MIN = 79, BuiltinOperator_FAKE_QUANT = 80, + BuiltinOperator_REDUCE_PROD = 81, + BuiltinOperator_REDUCE_MAX = 82, BuiltinOperator_MIN = BuiltinOperator_ADD, - BuiltinOperator_MAX = BuiltinOperator_FAKE_QUANT + BuiltinOperator_MAX = BuiltinOperator_REDUCE_MAX }; -inline BuiltinOperator (&EnumValuesBuiltinOperator())[80] { +inline BuiltinOperator (&EnumValuesBuiltinOperator())[82] { static BuiltinOperator values[] = { BuiltinOperator_ADD, BuiltinOperator_AVERAGE_POOL_2D, @@ -436,7 +438,9 @@ inline BuiltinOperator (&EnumValuesBuiltinOperator())[80] { BuiltinOperator_SHAPE, BuiltinOperator_POW, BuiltinOperator_ARG_MIN, - BuiltinOperator_FAKE_QUANT + BuiltinOperator_FAKE_QUANT, + BuiltinOperator_REDUCE_PROD, + BuiltinOperator_REDUCE_MAX }; return values; } @@ -524,6 +528,8 @@ inline const char **EnumNamesBuiltinOperator() { "POW", "ARG_MIN", "FAKE_QUANT", + "REDUCE_PROD", + "REDUCE_MAX", nullptr }; return names; |