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/model.cc | |
parent | c35bd2e9d3d9311bc7fb0f2463869faf1a8a7b50 (diff) |
Support narrow_range attr on FakeQuant nodes
PiperOrigin-RevId: 204339562
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
-rw-r--r-- | tensorflow/contrib/lite/model.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index 93b3df98f3..71e38c3f13 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -715,6 +715,7 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type, params->min = schema_params->min(); params->max = schema_params->max(); params->num_bits = schema_params->num_bits(); + params->narrow_range = schema_params->narrow_range(); } *builtin_data = static_cast<void*>(params); break; |