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 /eval/src/apps | |
parent | 234adeffe4f1e4f1275a834f42293722a2ce6829 (diff) |
run onnx model to detect unknown output sizes
Diffstat (limited to 'eval/src/apps')
-rw-r--r-- | eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp index ecf38f6772b..2f22f903f2e 100644 --- a/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp +++ b/eval/src/apps/analyze_onnx_model/analyze_onnx_model.cpp @@ -140,6 +140,7 @@ Onnx::WireInfo make_plan(Options &opts, const Onnx &model) { auto type = make_input_type(input); REQUIRE(planner.bind_input_type(type, input)); } + planner.prepare_output_types(model); for (const auto &output: model.outputs()) { REQUIRE(!planner.make_output_type(output).is_error()); } |