diff options
Diffstat (limited to 'src/mlia/target')
-rw-r--r-- | src/mlia/target/cortex_a/advisor.py | 6 | ||||
-rw-r--r-- | src/mlia/target/ethos_u/advisor.py | 4 | ||||
-rw-r--r-- | src/mlia/target/tosa/advisor.py | 6 |
3 files changed, 10 insertions, 6 deletions
diff --git a/src/mlia/target/cortex_a/advisor.py b/src/mlia/target/cortex_a/advisor.py index 1249d93..52af592 100644 --- a/src/mlia/target/cortex_a/advisor.py +++ b/src/mlia/target/cortex_a/advisor.py @@ -64,7 +64,7 @@ class CortexAInferenceAdvisor(DefaultInferenceAdvisor): def configure_and_get_cortexa_advisor( context: ExecutionContext, - target_profile: str, + target_profile: str | Path, model: str | Path, **_extra_args: Any, ) -> InferenceAdvisor: @@ -78,7 +78,9 @@ def configure_and_get_cortexa_advisor( return CortexAInferenceAdvisor() -def _get_config_parameters(model: str | Path, target_profile: str) -> dict[str, Any]: +def _get_config_parameters( + model: str | Path, target_profile: str | Path +) -> dict[str, Any]: """Get configuration parameters for the advisor.""" advisor_parameters: dict[str, Any] = { "cortex_a_inference_advisor": { diff --git a/src/mlia/target/ethos_u/advisor.py b/src/mlia/target/ethos_u/advisor.py index ce4e0fc..937e91c 100644 --- a/src/mlia/target/ethos_u/advisor.py +++ b/src/mlia/target/ethos_u/advisor.py @@ -123,7 +123,7 @@ class EthosUInferenceAdvisor(DefaultInferenceAdvisor): def configure_and_get_ethosu_advisor( context: ExecutionContext, - target_profile: str, + target_profile: str | Path, model: str | Path, **extra_args: Any, ) -> InferenceAdvisor: @@ -155,7 +155,7 @@ _DEFAULT_OPTIMIZATION_TARGETS = [ def _get_config_parameters( model: str | Path, - target_profile: str, + target_profile: str | Path, **extra_args: Any, ) -> dict[str, Any]: """Get configuration parameters for the advisor.""" diff --git a/src/mlia/target/tosa/advisor.py b/src/mlia/target/tosa/advisor.py index b60e824..e8aad53 100644 --- a/src/mlia/target/tosa/advisor.py +++ b/src/mlia/target/tosa/advisor.py @@ -78,7 +78,7 @@ class TOSAInferenceAdvisor(DefaultInferenceAdvisor): def configure_and_get_tosa_advisor( context: ExecutionContext, - target_profile: str, + target_profile: str | Path, model: str | Path, **_extra_args: Any, ) -> InferenceAdvisor: @@ -92,7 +92,9 @@ def configure_and_get_tosa_advisor( return TOSAInferenceAdvisor() -def _get_config_parameters(model: str | Path, target_profile: str) -> dict[str, Any]: +def _get_config_parameters( + model: str | Path, target_profile: str | Path +) -> dict[str, Any]: """Get configuration parameters for the advisor.""" advisor_parameters: dict[str, Any] = { "tosa_inference_advisor": { |