aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/devices/tosa/data_analysis.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/devices/tosa/data_analysis.py')
-rw-r--r--src/mlia/devices/tosa/data_analysis.py36
1 files changed, 0 insertions, 36 deletions
diff --git a/src/mlia/devices/tosa/data_analysis.py b/src/mlia/devices/tosa/data_analysis.py
deleted file mode 100644
index 7cbd61d..0000000
--- a/src/mlia/devices/tosa/data_analysis.py
+++ /dev/null
@@ -1,36 +0,0 @@
-# SPDX-FileCopyrightText: Copyright 2022, Arm Limited and/or its affiliates.
-# SPDX-License-Identifier: Apache-2.0
-"""TOSA data analysis module."""
-from dataclasses import dataclass
-from functools import singledispatchmethod
-
-from mlia.backend.tosa_checker.compat import TOSACompatibilityInfo
-from mlia.core.common import DataItem
-from mlia.core.data_analysis import Fact
-from mlia.core.data_analysis import FactExtractor
-
-
-@dataclass
-class ModelIsTOSACompatible(Fact):
- """Model is completely TOSA compatible."""
-
-
-@dataclass
-class ModelIsNotTOSACompatible(Fact):
- """Model is not TOSA compatible."""
-
-
-class TOSADataAnalyzer(FactExtractor):
- """TOSA data analyzer."""
-
- @singledispatchmethod
- def analyze_data(self, data_item: DataItem) -> None: # type: ignore
- """Analyse the data."""
-
- @analyze_data.register
- def analyze_tosa_compatibility(self, data_item: TOSACompatibilityInfo) -> None:
- """Analyse TOSA compatibility information."""
- if data_item.tosa_compatible:
- self.add_fact(ModelIsTOSACompatible())
- else:
- self.add_fact(ModelIsNotTOSACompatible())