diff options
Diffstat (limited to 'tests/test_target_cortex_a_operators.py')
-rw-r--r-- | tests/test_target_cortex_a_operators.py | 17 |
1 files changed, 0 insertions, 17 deletions
diff --git a/tests/test_target_cortex_a_operators.py b/tests/test_target_cortex_a_operators.py index 8bc48e6..56d6c7b 100644 --- a/tests/test_target_cortex_a_operators.py +++ b/tests/test_target_cortex_a_operators.py @@ -2,33 +2,16 @@ # SPDX-License-Identifier: Apache-2.0 """Tests for Cortex-A operator compatibility.""" from pathlib import Path -from typing import cast import pytest import tensorflow as tf -from mlia.backend.armnn_tflite_delegate import compat -from mlia.nn.tensorflow.tflite_graph import TFL_OP from mlia.nn.tensorflow.utils import convert_to_tflite from mlia.target.cortex_a.config import CortexAConfiguration from mlia.target.cortex_a.operators import CortexACompatibilityInfo from mlia.target.cortex_a.operators import get_cortex_a_compatibility_info -def test_compat_data() -> None: - """Make sure all data contains the necessary items.""" - builtin_tfl_ops = {op.name for op in TFL_OP} - assert "backend" in compat.ARMNN_TFLITE_DELEGATE - assert "ops" in compat.ARMNN_TFLITE_DELEGATE - - ops = cast(dict, compat.ARMNN_TFLITE_DELEGATE["ops"]) - for data in ops.values(): - assert "builtin_ops" in data - for comp in data["builtin_ops"]: - assert comp in builtin_tfl_ops - assert "custom_ops" in data - - def check_get_cortex_a_compatibility_info( model_path: Path, expected_success: bool, |