Exemplo n.º 1
0
  def testTargetOps(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(
        keras_file,
        target_ops=set([lite.OpsSet.TFLITE_BUILTINS,
                        lite.OpsSet.SELECT_TF_OPS]))
Exemplo n.º 2
0
  def testTargetOps(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(
        keras_file,
        target_ops=set([lite.OpsSet.TFLITE_BUILTINS,
                        lite.OpsSet.SELECT_TF_OPS]))
Exemplo n.º 3
0
    def testPostTrainingQuantize(self):
        model = self._getSingleInputKerasModel()
        keras_file = self._saveKerasModel(model)

        model_coverage.test_keras_model(keras_file,
                                        post_training_quantize=True)
Exemplo n.º 4
0
    def testFloat(self):
        model = self._getSingleInputKerasModel()
        keras_file = self._saveKerasModel(model)

        model_coverage.test_keras_model(keras_file)
Exemplo n.º 5
0
  def testPostTrainingQuantize(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(keras_file, post_training_quantize=True)
Exemplo n.º 6
0
  def testFloat(self):
    model = self._getSingleInputKerasModel()
    keras_file = self._saveKerasModel(model)

    model_coverage.test_keras_model(keras_file)