Python test_frozen_graph_quant Exemples

Exemple #1

 def testQuantizedInputShapes(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(
       filename, ['inputA', 'inputB'], ['output'],
       input_shapes={
           'inputA': [33, 33],
           'inputB': [33, 33],
       })

Exemple #2

 def testQuantizedFlexAll(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(
       filename, ['inputA', 'inputB'], ['output'],
       converter_mode=lite.ConverterMode.TOCO_FLEX_ALL)

Exemple #3

 def testQuantized(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(filename, ['inputA', 'inputB'],
                                          ['output'])

Exemple #4

 def testQuantizedFlexAll(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(
       filename, ['inputA'], ['output'],
       target_ops=set([lite.OpsSet.SELECT_TF_OPS]))