[IE SAMPLE] Fixed inputs and outputs element type initialization (#9375)
This commit is contained in:
parent
882f30cd61
commit
c4ce6c5430
@ -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();
|
||||
|
Loading…
Reference in New Issue
Block a user