From 45e653dbd81633b8d78215b16a9b2205e39dd8e2 Mon Sep 17 00:00:00 2001 From: Jonas Ohlsson Date: Mon, 26 Jul 2021 16:13:12 +0200 Subject: MLBEDSW-4853: Refactor supported operators Refactor supported operators by breaking out model semantics into its own class. Model semantics checked right after model read. Signed-off-by: Jonas Ohlsson Change-Id: If442b189efcd91dda01af60b2b3adedfacdf2fad --- ethosu/vela/architecture_features.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'ethosu/vela/architecture_features.py') diff --git a/ethosu/vela/architecture_features.py b/ethosu/vela/architecture_features.py index 98d3d8c2..aaf1ae45 100644 --- a/ethosu/vela/architecture_features.py +++ b/ethosu/vela/architecture_features.py @@ -32,12 +32,12 @@ from .numeric_util import round_up_to_int from .operation import Kernel from .operation import NpuBlockType from .operation import PointXYZ -from .supported_operators import SupportedOperators from .tensor import BandwidthDirection from .tensor import MemArea from .tensor import MemType from .tensor import TensorFormat from .tensor import TensorPurpose +from .tflite_supported_operators import TFLiteSupportedOperators from .tosa_supported_operators import TosaSupportedOperators @@ -398,7 +398,7 @@ class ArchitectureFeatures: self.generate_block_config_map(Block(ifm_block_max.width * 2, ifm_block_max.height, 128)) # Setup supported operators and restriction checkers class - self.supported_operators = SupportedOperators() + self.tflite_supported_operators = TFLiteSupportedOperators() self.tosa_supported_operators = TosaSupportedOperators() # Returns available number of SHRAM banks depending on activation lookup table -- cgit v1.2.1