diff options
author | 2018-07-10 10:22:57 -0700 | |
---|---|---|
committer | 2018-07-10 10:29:57 -0700 | |
commit | d2fa11acfd6b8a2a2663dc70cd899433bdde23e1 (patch) | |
tree | 55d9865cef5c19fec51556aa1b8b5afc66702efa /tensorflow/contrib/lite/model.cc | |
parent | a6df304e5ec4b0ad28884025558ecda414b9df07 (diff) |
Adding FakeQuant op support.
PiperOrigin-RevId: 203969441
Diffstat (limited to 'tensorflow/contrib/lite/model.cc')
-rw-r--r-- | tensorflow/contrib/lite/model.cc | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/model.cc b/tensorflow/contrib/lite/model.cc index ca4821e84d..93b3df98f3 100644 --- a/tensorflow/contrib/lite/model.cc +++ b/tensorflow/contrib/lite/model.cc @@ -709,6 +709,16 @@ TfLiteStatus ParseOpData(const Operator* op, BuiltinOperator op_type, error_reporter->Report("DELEGATE op shouldn't exist in model."); return kTfLiteError; } + case BuiltinOperator_FAKE_QUANT: { + auto* params = MallocPOD<TfLiteFakeQuantParams>(); + if (auto* schema_params = op->builtin_options_as_FakeQuantOptions()) { + params->min = schema_params->min(); + params->max = schema_params->max(); + params->num_bits = schema_params->num_bits(); + } + *builtin_data = static_cast<void*>(params); + break; + } // Below are the ops with no builtin_data strcture. case BuiltinOperator_BATCH_TO_SPACE_ND: |