aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs24
1 files changed, 23 insertions, 1 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 131970e449..3a01c504a5 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -1,5 +1,5 @@
//
-// Copyright © 2017,2019-2023 Arm Ltd and Contributors. All rights reserved.
+// Copyright © 2017,2019-2024 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//
@@ -64,6 +64,14 @@ enum ResizeMethod: byte {
Bilinear = 1,
}
+enum ScatterNdFunction: byte {
+ Update = 0,
+ Add = 1,
+ Sub = 2,
+ Max = 3,
+ Min = 4
+}
+
table TensorInfo {
dimensions:[uint];
dataType:DataType;
@@ -189,6 +197,7 @@ enum LayerType : uint {
ElementwiseBinary = 69,
ReverseV2 = 70,
Tile = 71,
+ ScatterNd = 72,
}
// Base layer table to be used as part of other layers
@@ -1066,6 +1075,18 @@ table TileLayer {
descriptor:TileDescriptor;
}
+table ScatterNdDescriptor {
+ m_Function:ScatterNdFunction = Update;
+ m_InputEnabled:bool = true;
+ m_Axis:int = 0;
+ m_AxisEnabled:bool = false;
+}
+
+table ScatterNdLayer {
+ base:LayerBase;
+ descriptor:ScatterNdDescriptor;
+}
+
union Layer {
ActivationLayer,
AdditionLayer,
@@ -1139,6 +1160,7 @@ union Layer {
ElementwiseBinaryLayer,
ReverseV2Layer,
TileLayer,
+ ScatterNdLayer,
}
table AnyLayer {