diff --git a/src/mldebug/layer_info.py b/src/mldebug/layer_info.py
index 329b567..f94e9a3 100644
--- a/src/mldebug/layer_info.py
+++ b/src/mldebug/layer_info.py
@@ -57,7 +57,9 @@ 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,
 }