1 files changed, 6 insertions, 6 deletions
diff --git a/schema/tosa.fbs b/schema/tosa.fbs
index daf6475..6e84b22 100644
--- a/schema/tosa.fbs
+++ b/schema/tosa.fbs
@@ -139,9 +139,9 @@ enum Op:uint32 {
 }
 
 union Attribute {
-  Pool2dAttribute,
-  Conv2dAttribute,
-  TransposeConv2dAttribute,
+  PoolAttribute,
+  ConvAttribute,
+  TransposeConvAttribute,
   ReluNAttribute,
   AxisAttribute,
   ReshapeAttribute,
@@ -156,19 +156,19 @@ union Attribute {
   WhileLoopAttribute,
 }
 
-table Pool2dAttribute {
+table PoolAttribute {
   padding: [int32];
   kernel: [int32];
   stride: [int32];
 }
 
-table Conv2dAttribute {
+table ConvAttribute {
   padding: [int32];
   stride: [int32];
   dilation: [int32];
 }
 
-table TransposeConv2dAttribute {
+table TransposeConvAttribute {
   outpad: [int32];
   stride: [int32];
   dilation: [int32];