mirror of
https://github.com/nod-ai/SHARK-Studio.git
synced 2026-01-10 06:17:55 -05:00
26 lines
670 B
Python
26 lines
670 B
Python
import absl.flags
|
|
import absl.testing
|
|
import test_util
|
|
|
|
absl.flags.DEFINE_string("model", None, "model path to execute")
|
|
|
|
|
|
class ManualTest(test_util.TFLiteModelTest):
|
|
def __init__(self, *args, **kwargs):
|
|
super(ManualTest, self).__init__(
|
|
absl.flags.FLAGS.model, *args, **kwargs
|
|
)
|
|
|
|
def compare_results(self, iree_results, tflite_results, details):
|
|
super(ManualTest, self).compare_results(
|
|
iree_results, tflite_results, details
|
|
)
|
|
|
|
def test_compile_tflite(self):
|
|
if self.model_path is not None:
|
|
self.compile_and_execute()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
absl.testing.absltest.main()
|