* fix key error in get_profiling_data
* make retrieval of inference measurements more generic
Signed-off-by: Nina Drozd <nina.drozd@arm.com>
Change-Id: I3fc147a6a93830b59e8b12f517be9f9c72370c09
top_level_dict = json.loads(profiler.as_json())
armnn_data = top_level_dict["ArmNN"]
- inference_measurements = armnn_data["inference_measurements_#1"]
- execution_data = inference_measurements["Execute_#2"]
+ #Get the inference measurements dict, this will be just one value for key starting with "inference_measurements"
+ inference_measurements = [v for k, v in armnn_data.items() if k.startswith("inference_measurements_")][0]
+
+ #Get the execution data dict, this will be just one value for key starting with "Execute_"
+ execution_data = [v for k, v in inference_measurements.items() if k.startswith("Execute_")][0]
workload_data = {}
inference_data = {}