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

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

View File

@ -89,6 +89,8 @@ int main(int argc, char* argv[]) {
ov::preprocess::PrePostProcessor proc(model);
for (int i = 0; i < model->inputs().size(); i++) {
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);
}
model = proc.build();