예제 #1
0
 def testQuantizedInputShapes(self):
     filename = self._getQuantizedModel()
     model_coverage.test_frozen_graph_quant(
         filename, ['inputA'], ['output'],
         input_shapes={'inputA': [33, 33]})
예제 #2
0
 def testQuantized(self):
     filename = self._getQuantizedModel()
     model_coverage.test_frozen_graph_quant(filename, ['inputA'],
                                            ['output'])
예제 #3
0
 def testQuantizedFlexAll(self):
     filename = self._getQuantizedModel()
     model_coverage.test_frozen_graph_quant(
         filename, ['inputA'], ['output'],
         target_ops=set([lite.OpsSet.SELECT_TF_OPS]))
예제 #4
0
 def testQuantizedInputShapes(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(
       filename, ['inputA'], ['output'], input_shapes={'inputA': [33, 33]})
예제 #5
0
 def testQuantized(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(filename, ['inputA'], ['output'])
예제 #6
0
 def testQuantizedFlexAll(self):
   filename = self._getQuantizedModel()
   model_coverage.test_frozen_graph_quant(
       filename, ['inputA'], ['output'],
       target_ops=set([lite.OpsSet.SELECT_TF_OPS]))