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/kernels/fake_quant.cc | |
parent | c35bd2e9d3d9311bc7fb0f2463869faf1a8a7b50 (diff) |
Support narrow_range attr on FakeQuant nodes
PiperOrigin-RevId: 204339562
Diffstat (limited to 'tensorflow/contrib/lite/kernels/fake_quant.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/fake_quant.cc | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/fake_quant.cc b/tensorflow/contrib/lite/kernels/fake_quant.cc index f8927a0799..0ef1a50b30 100644 --- a/tensorflow/contrib/lite/kernels/fake_quant.cc +++ b/tensorflow/contrib/lite/kernels/fake_quant.cc @@ -44,6 +44,17 @@ TfLiteStatus Prepare(TfLiteContext* context, TfLiteNode* node) { TF_LITE_ENSURE_EQ(context, NumInputs(node), 1); TF_LITE_ENSURE_EQ(context, NumOutputs(node), 1); + const auto* params = + reinterpret_cast<TfLiteFakeQuantParams*>(node->builtin_data); + + if (params->narrow_range) { + context->ReportError( + context, + "narrow_range FakeQuant is not currently supported at runtime. " + "narrow_range is only meant to be applied to weights, not activations"); + return kTfLiteError; + } + OpContext op_context(context, node); TfLiteIntArray* output_dims = TfLiteIntArrayCopy(op_context.input->dims); op_context.output->type = op_context.input->type; |