diff options
author | Håvard Pettersen <havardpe@oath.com> | 2022-02-08 10:59:07 +0000 |
---|---|---|
committer | Håvard Pettersen <havardpe@oath.com> | 2022-02-09 14:45:48 +0000 |
commit | e4dd5bce73e2ac856d359098c27195e5aebbbf8b (patch) | |
tree | aaa0fc73c83258b3d02e26445464df7b11bacc7a /searchlib/src | |
parent | 234adeffe4f1e4f1275a834f42293722a2ce6829 (diff) |
run onnx model to detect unknown output sizes
Diffstat (limited to 'searchlib/src')
-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); |