diff options
author | Lester Solbakken <lesters@users.noreply.github.com> | 2022-02-10 09:16:25 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2022-02-10 09:16:25 +0100 |
commit | deeef73ce7e42e26f325b4bb2d6f05112b31ff70 (patch) | |
tree | 90436ab43332c8666c68a57d7334d2481b295163 /searchlib | |
parent | 0b284f194c0b2dc9b1f8b36b489911f32961d840 (diff) | |
parent | 8eae2d45b4f03eb8f696bc1c13ba0a5768a38f12 (diff) |
Merge pull request #21124 from vespa-engine/havardpe/run-onnx-model-to-detect-output-sizes
run onnx model to detect unknown output sizes
Diffstat (limited to 'searchlib')
-rw-r--r-- | searchlib/src/vespa/searchlib/features/onnx_feature.cpp | 1 |
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); |