diff options
| author |  Jared Duke <jdduke@google.com> | 2018-07-26 10:53:21 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-26 10:56:43 -0700 |
| commit | 6e658c0a5ca77677a954a34fb98f241c592c970d (patch) | |
| tree | b645103887539af5232b3f70d80a2eb9b77ed63a /tensorflow/contrib/lite/schema/schema.fbs | |
| parent | 0a3155f7fbf56df5e81c7cbf35afd45173359635 (diff) | |
Add one_hot op support to TFLite
PiperOrigin-RevId: 206185190
Diffstat (limited to 'tensorflow/contrib/lite/schema/schema.fbs')
| -rw-r--r-- | tensorflow/contrib/lite/schema/schema.fbs | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/schema/schema.fbs b/tensorflow/contrib/lite/schema/schema.fbs index a285bf9919..8ed98ddaf4 100644 --- a/tensorflow/contrib/lite/schema/schema.fbs +++ b/tensorflow/contrib/lite/schema/schema.fbs @@ -166,6 +166,7 @@ enum BuiltinOperator : byte { REDUCE_MAX = 82, PACK = 83, LOGICAL_OR = 84, + ONE_HOT = 85, } // Options for the builtin operators. @@ -230,6 +231,7 @@ union BuiltinOptions { FakeQuantOptions, PackOptions, LogicalOrOptions, + OneHotOptions, } enum Padding : byte { SAME, VALID } @@ -549,6 +551,10 @@ table PackOptions { table LogicalOrOptions { } +table OneHotOptions { + axis:int; +} + // An OperatorCode can be an enum value (BuiltinOperator) if the operator is a // builtin, or a string if the operator is custom. table OperatorCode { |