diff options
Diffstat (limited to 'python/pyarmnn/src/pyarmnn/__init__.py')
| -rw-r--r-- | python/pyarmnn/src/pyarmnn/__init__.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/python/pyarmnn/src/pyarmnn/__init__.py b/python/pyarmnn/src/pyarmnn/__init__.py index b1aa81f507..19b14a4b09 100644 --- a/python/pyarmnn/src/pyarmnn/__init__.py +++ b/python/pyarmnn/src/pyarmnn/__init__.py @@ -59,6 +59,18 @@ except ImportError as err: """In case people try importing without having Arm NN built with this parser.""" raise RuntimeError(message) +try: + from ._generated.pyarmnn_deserializer import IDeserializer +except ImportError as err: + logger = logging.getLogger(__name__) + message = "Your ArmNN library instance does not have an armnn models parser funcionality. " + logger.warning("%s Skipped IDeserializer import.", message) + logger.debug(str(err)) + + def IDeserializer(): + """In case people try importing without having ArmNN built with this parser.""" + raise RuntimeError(message) + # Network from ._generated.pyarmnn import Optimize, OptimizerOptions, IOptimizedNetwork, IInputSlot, \ IOutputSlot, IConnectableLayer, INetwork |