diff options
author | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-24 15:08:08 +0100 |
---|---|---|
committer | Dmitrii Agibov <dmitrii.agibov@arm.com> | 2022-10-26 17:08:13 +0100 |
commit | 58a65fee574c00329cf92b387a6d2513dcbf6100 (patch) | |
tree | 47e3185f78b4298ab029785ddee68456e44cac10 /src/mlia/devices/cortexa/advisor.py | |
parent | 9d34cb72d45a6d0a2ec1063ebf32536c1efdba75 (diff) | |
download | mlia-58a65fee574c00329cf92b387a6d2513dcbf6100.tar.gz |
MLIA-433 Add TensorFlow Lite compatibility check
- Add ability to intercept low level TensorFlow output
- Produce advice for the models that could not be
converted to the TensorFlow Lite format
- Refactor utility functions for TensorFlow Lite
conversion
- Add TensorFlow Lite compatibility checker
Change-Id: I47d120d2619ced7b143bc92c5184515b81c0220d
Diffstat (limited to 'src/mlia/devices/cortexa/advisor.py')
-rw-r--r-- | src/mlia/devices/cortexa/advisor.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/src/mlia/devices/cortexa/advisor.py b/src/mlia/devices/cortexa/advisor.py index 98c155b..ffbbea5 100644 --- a/src/mlia/devices/cortexa/advisor.py +++ b/src/mlia/devices/cortexa/advisor.py @@ -68,16 +68,14 @@ def configure_and_get_cortexa_advisor( target_profile: str, model: str | Path, output: PathOrFileLike | None = None, - **extra_args: Any, + **_extra_args: Any, ) -> InferenceAdvisor: """Create and configure Cortex-A advisor.""" if context.event_handlers is None: context.event_handlers = [CortexAEventHandler(output)] if context.config_parameters is None: - context.config_parameters = _get_config_parameters( - model, target_profile, **extra_args - ) + context.config_parameters = _get_config_parameters(model, target_profile) return CortexAInferenceAdvisor() |