diff options
Diffstat (limited to 'src/mlia/devices/cortexa/operators.py')
-rw-r--r-- | src/mlia/devices/cortexa/operators.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/src/mlia/devices/cortexa/operators.py b/src/mlia/devices/cortexa/operators.py new file mode 100644 index 0000000..6a314b7 --- /dev/null +++ b/src/mlia/devices/cortexa/operators.py @@ -0,0 +1,29 @@ +# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates. +# SPDX-License-Identifier: Apache-2.0 +"""Cortex-A tools module.""" +from __future__ import annotations + +from dataclasses import dataclass +from pathlib import Path + + +@dataclass +class Operator: + """Cortex-A compatibility information of the operator.""" + + name: str + location: str + is_cortex_a_compatible: bool + + +@dataclass +class CortexACompatibilityInfo: + """Model's operators.""" + + cortex_a_compatible: bool + operators: list[Operator] + + +def get_cortex_a_compatibility_info(model_path: Path) -> CortexACompatibilityInfo: + """Return list of model's operators.""" + raise NotImplementedError() |