IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[IE SAMPLE] Fixed inputs and outputs element type initialization (#9375)

Browse Source
This commit is contained in:
Elizaveta Lobanova 2021-12-23 12:06:23 +03:00 committed by GitHub
parent 882f30cd61
commit c4ce6c5430
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
samples/cpp/speech_sample/main.cpp
View File

@ -89,6 +89,8 @@ int main(int argc, char* argv[]) {
ov::preprocess::PrePostProcessor proc(model); ov::preprocess::PrePostProcessor proc(model);
for (int i = 0; i < model->inputs().size(); i++) { for (int i = 0; i < model->inputs().size(); i++) {
proc.input(i).tensor().set_element_type(ov::element::f32).set_layout(tensor_layout); proc.input(i).tensor().set_element_type(ov::element::f32).set_layout(tensor_layout);
}
for (int i = 0; i < model->outputs().size(); i++) {
proc.output(i).tensor().set_element_type(ov::element::f32); proc.output(i).tensor().set_element_type(ov::element::f32);
} }
model = proc.build(); model = proc.build();