From 249304793c2e13b219ded6f939c57a0e8557fa0e Mon Sep 17 00:00:00 2001
From: Brian Jones <brian.j.jones@intel.com>
Date: Mon, 21 Nov 2022 10:53:16 -0800
Subject: [PATCH] Minor fix to correctly calculate F16 buffer size

---
 image2tensor/src/image2tensor.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/image2tensor/src/image2tensor.rs b/image2tensor/src/image2tensor.rs
index 0685597..d09aa53 100644
--- a/image2tensor/src/image2tensor.rs
+++ b/image2tensor/src/image2tensor.rs
@@ -76,7 +76,7 @@ fn save_bytes(arr: &[u8], tt: TensorType) -> Vec<u8> {
 fn get_bytes_per_pixel(precision: TensorType) -> usize {
     match precision {
         TensorType::F32 | TensorType::I32 => 4,
-        TensorType::F16 => 2,
+        TensorType::F16 => 4,   // Currently Rust doesn't support F16 natively, so we use f32.
         TensorType::U8 => 1,
     }
 }