diff --git a/src/mldebug/layer_info.py b/src/mldebug/layer_info.py
index 54ae7d7..a6c5e15 100644
--- a/src/mldebug/layer_info.py
+++ b/src/mldebug/layer_info.py
@@ -34,6 +34,7 @@
   "mllib_graphs::topk_adf_wrapper",
   "mllib_graphs::transpose4d_adf_wrapper<signed char>",
   "mllib_graphs::transpose4d_adf_wrapper<bfloat16>",
+  "mllib_graphs::transpose4d_adf_wrapper<int8>",
 ]
 
 # Kernels that work on PHX/STX but are unsupported on other devices.