diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-08-03 12:26:54 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-03 12:31:18 -0700 |
commit | 69b4ee4c78c2551d027fd39e81fc71e1a6698f31 (patch) | |
tree | 7dce4fe0d8904b4ed30e35e8ec355507703259f7 /tensorflow/contrib/lite/toco/model.h | |
parent | dd45704d092dac87575b8ce39013f91f4f213dc0 (diff) |
Add support for float output arrays in Quantized custom operators (custom ops only).
PiperOrigin-RevId: 207306198
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
-rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index cd263930f5..18c78e32d0 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -1532,6 +1532,9 @@ struct TensorFlowUnsupportedOperator : Operator { string tensorflow_node_def; // A boolean indicating if the unsupported op should be treated as quantized. bool quantized = false; + // A boolean indicating if the unsupported op output should allow float values + // in quantized mode. + bool support_output_type_float_in_quantized_op = false; // Output data types std::vector<ArrayDataType> output_data_types; // Output shapes. |