STT-tensorflow/tensorflow/lite/tools/optimize/testdata
Suharsh Sivakumar 9e668b3d69 Add support for partially quantizing models.
PiperOrigin-RevId: 245479701
2019-04-26 14:38:08 -07:00
..
add_with_const_input.bin Handle constant inputs in quantizer. 2019-03-15 16:50:55 -07:00
argmax.bin Add support for Argmax in quantize_model. 2019-04-14 21:01:13 -07:00
concat.bin Support Concat in quantizer. 2019-04-04 09:16:31 -07:00
custom_op.bin Add api for weight quantization of custom operations. 2019-04-08 12:00:49 -07:00
fc.bin Support quantized FC layer (with biases). 2019-04-15 16:59:28 -07:00
mixed.bin Add support for partially quantizing models. 2019-04-26 14:38:08 -07:00
multi_input_add_reshape.bin Add Reshape and Add support to quantizer. 2019-02-28 11:05:51 -08:00
quantized_with_gather.bin Support for post-training quantization for Gather 2019-03-20 10:33:16 -07:00
README.md Support for post-training quantization for Gather 2019-03-20 10:33:16 -07:00
single_avg_pool_min_minus_5_max_plus_5.bin Enable opensource tests. 2019-01-25 14:36:38 -08:00
single_conv_weights_min_0_max_plus_10.bin Enable opensource tests. 2019-01-25 14:36:38 -08:00
single_conv_weights_min_minus_127_max_plus_127.bin Enable opensource tests. 2019-01-25 14:36:38 -08:00
single_softmax_min_minus_5_max_plus_5.bin Enable opensource tests. 2019-01-25 14:36:38 -08:00
weight_shared_between_convs.bin Enable opensource tests. 2019-01-25 14:36:38 -08:00

Test models for testing quantization

This directory contains test models for testing quantization.

Models

  • single_conv_weights_min_0_max_plus_10.bin
    A floating point model with single convolution where all weights are integers between [0, 10] weights are randomly distributed. It is not guaranteed that min max for weights are going to appear in each channel. All activations have min maxes and activations are in range [0,10].
  • single_conv_weights_min_minus_127_max_plus_127.bin
    A floating point model with a single convolution where weights of the model are all integers that lie in range[-127, 127]. The weights have been put in such a way that each channel has at least one weight as -127 and one weight as 127. The activations are all in range: [-128, 127]. This means all bias computations should result in 1.0 scale.
  • single_softmax_min_minus_5_max_5.bin
    A floating point model with a single softmax. The input tensor has min and max in range [-5, 5], not necessarily -5 or +5.
  • single_avg_pool_input_min_minus_5_max_5.bin
    A floating point model with a single average pool. The input tensor has min and max in range [-5, 5], not necessarily -5 or +5.
  • weight_shared_between_convs.bin
    A floating point model with two convs that have a use the same weight tensor.
  • multi_input_add_reshape.bin
    A floating point model with two inputs with an add followed by a reshape.
  • quantized_with_gather.bin
    A floating point model with an input with a gather, modeling a situation of mapping categorical input to embeddings.