diff options
Diffstat (limited to 'src/mlia/devices/tosa/data_analysis.py')
-rw-r--r-- | src/mlia/devices/tosa/data_analysis.py | 36 |
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()) |