From cea0885ab26b50beaf2da7a7cb694f30bf60b313 Mon Sep 17 00:00:00 2001
From: Rickard Bolin <rickard.bolin@arm.com>
Date: Fri, 2 May 2025 11:18:08 +0000
Subject: [PATCH] MLBEDSW-10747: Fix input tensor check in TOSA reader

Reader expected all operators to have input tensors, but CONST operator
does not.

Change-Id: I696d9a36cbc179988ded61f9b020a46b40df89ef
Signed-off-by: Rickard Bolin <rickard.bolin@arm.com>
---
 ethosu/regor/tosa/tosa_reader.cpp | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/ethosu/regor/tosa/tosa_reader.cpp b/ethosu/regor/tosa/tosa_reader.cpp
index 2fb15473..902dde18 100644
--- a/ethosu/regor/tosa/tosa_reader.cpp
+++ b/ethosu/regor/tosa/tosa_reader.cpp
@@ -373,11 +373,14 @@ void TosaReader::LoadGraphs(const tosaFb::TosaGraph *model, std::list<GraphBuild
             {
                 const auto &tosa_operator = SafeDeref(tosa_operators[tosa_op_index]);
                 // Connect operation to its input tensors
-                const auto &input_tensors_fb = SafeDeref(tosa_operator.inputs());
                 std::vector<std::string> input_tensors;
-                input_tensors.reserve(input_tensors_fb.size());
-                for ( const auto &ten : input_tensors_fb )
-                    input_tensors.push_back(SafeDeref(ten).str());
+                if ( tosa_operator.inputs() )
+                {
+                    const auto &input_tensors_fb = SafeDeref(tosa_operator.inputs());
+                    input_tensors.reserve(input_tensors_fb.size());
+                    for ( const auto &ten : input_tensors_fb )
+                        input_tensors.push_back(SafeDeref(ten).str());
+                }
                 const auto &output_tensors = SafeDeref(tosa_operator.outputs());
 
                 // Kernel
-- 
GitLab