aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/lite/builtin_op_data.h
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-07-12 12:04:34 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-07-12 12:08:17 -0700
commit9ba6943a1dbbc415b72835517ad58808ca6a6a3d (patch)
tree8c40f84f8dcb6deb9799664a0c8068822c6897b9 /tensorflow/contrib/lite/builtin_op_data.h
parentc35bd2e9d3d9311bc7fb0f2463869faf1a8a7b50 (diff)
Support narrow_range attr on FakeQuant nodes
PiperOrigin-RevId: 204339562
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r--tensorflow/contrib/lite/builtin_op_data.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index a58dde9a7b..a24aaad7dd 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -268,9 +268,13 @@ typedef struct {
} TfLiteShapeParams;
typedef struct {
+ // Parameters supported by version 1:
float min;
float max;
int num_bits;
+
+ // Parameters supported by version 2:
+ bool narrow_range;
} TfLiteFakeQuantParams;
#ifdef __cplusplus