diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-04-10 16:26:05 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-10 16:28:49 -0700 |
| commit | bb1dec54a63ad0a5f43208fa7617f090bc5be2e9 (patch) | |
| tree | 4aad9d5f6b859df8089a2c18a4daa4d942abac28 /tensorflow/contrib/lite/toco/model.h | |
| parent | 2891e0930eba15c7f27b0ab5732554e6b2c474d5 (diff) | |
Supporting FakeQuant num_bits and getting the fake quant op matching tensorflow.
PiperOrigin-RevId: 192367307
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 54c3a59506..616d53ae3e 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -724,8 +724,7 @@ struct L2PoolOperator : Operator { // The expected [min, max] range of values in a given array. // Used for quantization only. // This information typically comes from special nodes found in quantized -// models, -// see FakeQuantOperator, and is used during quantization to resolve +// models, see FakeQuantOperator, and is used during quantization to resolve // actual quantization parameters (see QuantizationParams). struct MinMax { double min = 0.; @@ -753,6 +752,7 @@ inline bool operator==(const MinMax& m1, const MinMax& m2) { struct FakeQuantOperator : Operator { FakeQuantOperator() : Operator(OperatorType::kFakeQuant) {} std::unique_ptr<MinMax> minmax; + int num_bits = 8; }; // Element-wise division operator. |