benchmark_app: fix FP16 image type (#15393)
This commit is contained in:
parent
85597f866c
commit
a19d50d4d2
@ -216,12 +216,12 @@ ov::Tensor get_image_tensor(const std::vector<std::string>& files,
|
||||
inputInfo.first,
|
||||
filenames_used);
|
||||
} else if (type == ov::element::f16) {
|
||||
return create_tensor_from_image<short>(files,
|
||||
inputId,
|
||||
batchSize,
|
||||
inputInfo.second,
|
||||
inputInfo.first,
|
||||
filenames_used);
|
||||
return create_tensor_from_image<ov::float16>(files,
|
||||
inputId,
|
||||
batchSize,
|
||||
inputInfo.second,
|
||||
inputInfo.first,
|
||||
filenames_used);
|
||||
} else if (type == ov::element::f64) {
|
||||
return create_tensor_from_image<double>(files,
|
||||
inputId,
|
||||
|
Loading…
Reference in New Issue
Block a user