diff --git a/ethosu/regor/tosa/tosa_reader.cpp b/ethosu/regor/tosa/tosa_reader.cpp index 2fb154733657eaf42fb0e0dfe12f3d7d94106c6a..902dde18137fe039b0c769f9e12de8559a5d8573 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 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