From 5f960d92dbefabf708e8f299789b35f6cdf2d919 Mon Sep 17 00:00:00 2001 From: wangg Date: Wed, 26 Aug 2020 01:44:32 +0000 Subject: Add IDeserializer support in pyarmnn Resources required for new unit tests are included for review. Signed-off-by: Guanqun Wang gemini910621@gmail.com Change-Id: Iead6cb5beaf824a6f467ad9da4aede5719ebe4ec --- python/pyarmnn/test/test_generated.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) (limited to 'python/pyarmnn/test/test_generated.py') diff --git a/python/pyarmnn/test/test_generated.py b/python/pyarmnn/test/test_generated.py index 24765c73ab..0e1d663234 100644 --- a/python/pyarmnn/test/test_generated.py +++ b/python/pyarmnn/test/test_generated.py @@ -1,4 +1,4 @@ -# Copyright © 2020 Arm Ltd. All rights reserved. +# Copyright © 2020 Arm Ltd and Contributors. All rights reserved. # SPDX-License-Identifier: MIT import inspect from typing import Tuple @@ -6,6 +6,7 @@ from typing import Tuple import pytest import pyarmnn._generated.pyarmnn as generated_armnn +import pyarmnn._generated.pyarmnn as generated_deserializer import pyarmnn._generated.pyarmnn_caffeparser as generated_caffe import pyarmnn._generated.pyarmnn_onnxparser as generated_onnx import pyarmnn._generated.pyarmnn_tfliteparser as generated_tflite @@ -26,6 +27,7 @@ def get_classes(swig_independent_classes: Tuple): ignored_class_names = (*swig_independent_classes, '_SwigNonDynamicMeta') return list(filter(lambda x: x[0] not in ignored_class_names, inspect.getmembers(generated_armnn, inspect.isclass) + + inspect.getmembers(generated_deserializer, inspect.isclass) + inspect.getmembers(generated_caffe, inspect.isclass) + inspect.getmembers(generated_tflite, inspect.isclass) + inspect.getmembers(generated_onnx, inspect.isclass) + -- cgit v1.2.1