Merge pull request #21124 from vespa-engine/havardpe/run-onnx-model-to-detect-output-sizes

run onnx model to detect unknown output sizes

1 files changed, 1 insertions, 0 deletions

diff --git a/searchlib/src/vespa/searchlib/features/onnx_feature.cpp b/searchlib/src/vespa/searchlib/features/onnx_feature.cpp
index c28abe10b4a..cdeb0515659 100644
--- a/searchlib/src/vespa/searchlib/features/onnx_feature.cpp
+++ b/searchlib/src/vespa/searchlib/features/onnx_feature.cpp
@@ -150,6 +150,7 @@ OnnxBlueprint::setup(const IIndexEnvironment &env,
             return fail("undefined input: %s (->%s)", input_feature.value().c_str(), model_input.name.c_str());
         }
     }
+    planner.prepare_output_types(*_model);
     for (size_t i = 0; i < _model->outputs().size(); ++i) {
         const auto &model_output = _model->outputs()[i];
         auto output_name = model_cfg->output_name(model_output.name);