postrpocessing: imx500: Update MAX_NUM_TENSORS and MAX_NUM_DIMENSIONS

Update these values to 16 to match the new libcamera definitions. Signed-off-by: Naushir Patuck <[email protected]>
raspberrypi · Nov 12, 2024 · 5c5da49 · 5c5da49
1 parent 225bef2
commit 5c5da49
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/post_processing_stages/imx500/imx500_post_processing_stage.hpp b/post_processing_stages/imx500/imx500_post_processing_stage.hpp
@@ -22,15 +22,15 @@
 class IMX500PostProcessingStage : public PostProcessingStage
 {
 public:
-	static constexpr unsigned int Max_Num_Tensors = 8;
+	static constexpr unsigned int Max_Num_Tensors = 16;
+	static constexpr unsigned int Max_Num_Dimensions = 16;
 	static constexpr unsigned int Network_Name_Len = 64;
 
 	struct OutputTensorInfo
 	{
 		uint32_t tensor_data_num;
-		uint16_t size;
-		uint8_t ordinal;
-		uint8_t serialization_index;
+		uint32_t num_dimensions;
+		uint16_t size[Max_Num_Dimensions];
 	};
 
 	struct CnnOutputTensorInfo