diff options
author | 2018-07-12 12:04:34 -0700 | |
---|---|---|
committer | 2018-07-12 12:08:17 -0700 | |
commit | 9ba6943a1dbbc415b72835517ad58808ca6a6a3d (patch) | |
tree | 8c40f84f8dcb6deb9799664a0c8068822c6897b9 /tensorflow/contrib/lite/builtin_op_data.h | |
parent | c35bd2e9d3d9311bc7fb0f2463869faf1a8a7b50 (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.h | 4 |
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 |