aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/ArmnnSchema.fbs
diff options
context:
space:
mode:
authorjosh minor <josh.minor@arm.com>2020-01-06 16:40:46 -0600
committerDerek Lamberti <derek.lamberti@arm.com>2020-01-23 14:29:14 +0000
commit4a3c61091037e7e86e8b03bb060d8c1ab82731a9 (patch)
tree928644023400ad5ac0c26b33dfff2f975567d6e8 /src/armnnSerializer/ArmnnSchema.fbs
parent190a39a4a9598e42b636ae4ab843761884148160 (diff)
downloadarmnn-4a3c61091037e7e86e8b03bb060d8c1ab82731a9.tar.gz
IVGCVSW-4259 Add frontend and reference workload for UnaryOperationLayer
* Added new layer named ElementwiseUnary * Deprecated existing Abs/Rsqrt layer functions * Updated existing Abs/Rsqrt test infrastructure to use new layer * Added boilerplate for new Exp,Neg,Sqrt elemwise op layers * AbsQuantize test removed pending future commit * Serialization support added !android-nn-driver:2550 Change-Id: Ic595c645925e17b45db568187fd05646daf2e87f Signed-off-by: josh minor <josh.minor@arm.com>
Diffstat (limited to 'src/armnnSerializer/ArmnnSchema.fbs')
-rw-r--r--src/armnnSerializer/ArmnnSchema.fbs25
1 files changed, 23 insertions, 2 deletions
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 0d30d96452..0f8a816093 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -147,7 +147,8 @@ enum LayerType : uint {
InstanceNormalization = 50,
LogSoftmax = 51,
Comparison = 52,
- StandIn = 53
+ StandIn = 53,
+ ElementwiseUnary = 54
}
// Base layer table to be used as part of other layers
@@ -166,6 +167,7 @@ table BindableLayerBase {
// Table for each layer defined below
+/// @deprecated Use ElementwiseUnaryLayer instead
table AbsLayer {
base:LayerBase;
}
@@ -252,6 +254,23 @@ table DivisionLayer {
base:LayerBase;
}
+enum UnaryOperation : byte {
+ Abs = 0,
+ Rsqrt = 1,
+ Sqrt = 2,
+ Exp = 3,
+ Neg = 4
+}
+
+table ElementwiseUnaryDescriptor {
+ operation:UnaryOperation;
+}
+
+table ElementwiseUnaryLayer {
+ base:LayerBase;
+ descriptor:ElementwiseUnaryDescriptor;
+}
+
/// @deprecated Use ComparisonLayer instead
table EqualLayer {
base:LayerBase;
@@ -502,6 +521,7 @@ table PadDescriptor {
padValue:float = 0;
}
+/// @deprecated Use ElementwiseUnaryLayer instead
table RsqrtLayer {
base:LayerBase;
}
@@ -798,7 +818,8 @@ union Layer {
InstanceNormalizationLayer,
LogSoftmaxLayer,
ComparisonLayer,
- StandInLayer
+ StandInLayer,
+ ElementwiseUnaryLayer
}
table AnyLayer {