From 98d63f25c32f297532072b91fbe4f7d20c90e3f3 Mon Sep 17 00:00:00 2001
From: Snigdha Gupta <71372521+snigdha-gupta@users.noreply.github.com>
Date: Tue, 1 Sep 2026 14:05:08 -0700
Subject: [PATCH 1/2]  Add uint16_t to SIZE_BYTES

Signed-off-by: Snigdha Gupta <71372521+snigdha-gupta@users.noreply.github.com>
---
 src/mldebug/layer_info.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mldebug/layer_info.py b/src/mldebug/layer_info.py
index 329b567..5ef8106 100644
--- a/src/mldebug/layer_info.py
+++ b/src/mldebug/layer_info.py
@@ -58,6 +58,7 @@ def _strip_template(name):
   "int8_t": 1,
   "uint8_t": 1,
   "uint16": 1,  # Size of x2 designs is already in bytes
+  "uint16_t": 2,
 }
 
 

From 52863f23c31472d542e23c886d0517a5c99660dd Mon Sep 17 00:00:00 2001
From: Snigdha Gupta <71372521+snigdha-gupta@users.noreply.github.com>
Date: Tue, 1 Sep 2026 14:15:44 -0700
Subject: [PATCH 2/2] Add int16_t to SIZE_BYTES

Signed-off-by: Snigdha Gupta <71372521+snigdha-gupta@users.noreply.github.com>
---
 src/mldebug/layer_info.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/mldebug/layer_info.py b/src/mldebug/layer_info.py
index 5ef8106..f94e9a3 100644
--- a/src/mldebug/layer_info.py
+++ b/src/mldebug/layer_info.py
@@ -57,6 +57,7 @@ def _strip_template(name):
   "float16": 2,
   "int8_t": 1,
   "uint8_t": 1,
+  "int16_t": 2,
   "uint16": 1,  # Size of x2 designs is already in bytes
   "uint16_t": 2,
 }