aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/target
diff options
context:
space:
mode:
Diffstat (limited to 'src/mlia/target')
-rw-r--r--src/mlia/target/cortex_a/advisor.py6
-rw-r--r--src/mlia/target/ethos_u/advisor.py4
-rw-r--r--src/mlia/target/tosa/advisor.py6
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": {