diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index 057933c..f101fa3 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -36,6 +36,7 @@
   UINT16,
   FP16,
   BF16,
+  SHAPE,
 }
 
 enum ResizeMode:uint32 {
@@ -117,6 +118,7 @@
   FFT2D,
   RFFT2D,
   ERF,
+  DIM,
 }
 
 union Attribute {