aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/schema/schema.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/lite/schema/schema.fbs')
-rw-r--r--tensorflow/contrib/lite/schema/schema.fbs36
1 files changed, 34 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/schema/schema.fbs b/tensorflow/contrib/lite/schema/schema.fbs
index 15fb8bbdb8..a285bf9919 100644
--- a/tensorflow/contrib/lite/schema/schema.fbs
+++ b/tensorflow/contrib/lite/schema/schema.fbs
@@ -44,7 +44,7 @@ enum TensorType : byte {
table QuantizationParameters {
min:[float]; // For importing back into tensorflow.
max:[float]; // For importing back into tensorflow.
- scale:[float];
+ scale:[float]; // For dequantizing the tensor's values.
zero_point:[long];
}
@@ -155,11 +155,17 @@ 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,
+ PACK = 83,
+ LOGICAL_OR = 84,
}
// Options for the builtin operators.
@@ -220,6 +226,10 @@ union BuiltinOptions {
NotEqualOptions,
ShapeOptions,
PowOptions,
+ ArgMinOptions,
+ FakeQuantOptions,
+ PackOptions,
+ LogicalOrOptions,
}
enum Padding : byte { SAME, VALID }
@@ -469,6 +479,10 @@ table ArgMaxOptions {
output_type : TensorType;
}
+table ArgMinOptions {
+ output_type : TensorType;
+}
+
table GreaterOptions {
}
@@ -517,6 +531,24 @@ table ShapeOptions {
table PowOptions {
}
+table FakeQuantOptions {
+ // Parameters supported by version 1:
+ min:float;
+ max:float;
+ num_bits:int;
+
+ // Parameters supported by version 2:
+ narrow_range:bool;
+}
+
+table PackOptions {
+ values_count:int;
+ axis:int;
+}
+
+table LogicalOrOptions {
+}
+
// An OperatorCode can be an enum value (BuiltinOperator) if the operator is a
// builtin, or a string if the operator is custom.
table OperatorCode {