aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/tosa/operators.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/devices/tosa/operators.py')
-rw-r--r--src/mlia/devices/tosa/operators.py66
1 files changed, 0 insertions, 66 deletions
diff --git a/src/mlia/devices/tosa/operators.py b/src/mlia/devices/tosa/operators.py
index 1e4581a..b75ceb0 100644
--- a/src/mlia/devices/tosa/operators.py
+++ b/src/mlia/devices/tosa/operators.py
@@ -1,72 +1,6 @@
# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
"""Operators module."""
-from __future__ import annotations
-
-from dataclasses import dataclass
-from typing import Any
-from typing import cast
-from typing import Protocol
-
-from mlia.core.typing import PathOrFileLike
-
-
-class TOSAChecker(Protocol):
- """TOSA checker protocol."""
-
- def is_tosa_compatible(self) -> bool:
- """Return true if model is TOSA compatible."""
-
- def _get_tosa_compatibility_for_ops(self) -> list[Any]:
- """Return list of operators."""
-
-
-@dataclass
-class Operator:
- """Operator's TOSA compatibility info."""
-
- location: str
- name: str
- is_tosa_compatible: bool
-
-
-@dataclass
-class TOSACompatibilityInfo:
- """Models' TOSA compatibility information."""
-
- tosa_compatible: bool
- operators: list[Operator]
-
-
-def get_tosa_compatibility_info(
- tflite_model_path: PathOrFileLike,
-) -> TOSACompatibilityInfo:
- """Return list of the operators."""
- checker = get_tosa_checker(tflite_model_path)
-
- if checker is None:
- raise Exception(
- "TOSA checker is not available. "
- "Please make sure that 'tosa-checker' backend is installed."
- )
-
- ops = [
- Operator(item.location, item.name, item.is_tosa_compatible)
- for item in checker._get_tosa_compatibility_for_ops() # pylint: disable=protected-access
- ]
-
- return TOSACompatibilityInfo(checker.is_tosa_compatible(), ops)
-
-
-def get_tosa_checker(tflite_model_path: PathOrFileLike) -> TOSAChecker | None:
- """Return instance of the TOSA checker."""
- try:
- import tosa_checker as tc # pylint: disable=import-outside-toplevel
- except ImportError:
- return None
-
- checker = tc.TOSAChecker(str(tflite_model_path))
- return cast(TOSAChecker, checker)
def report() -> None: