IVGCVSW-5410 Add front-end support for CAST
IVGCVSW-5415 Add TfLiteParser support for CAST
* Added front end support for CAST, including support in the
Reference workload, Serialization, Deserializtion, Unit tests, and
TfLiteParser.
Signed-off-by: mathad01 <matthew.haddon@arm.com>
Change-Id: Iaf670ca5912a21ed6bc84f7f83a68b42154846bb
diff --git a/src/armnnSerializer/ArmnnSchema.fbs b/src/armnnSerializer/ArmnnSchema.fbs
index 438ea83..a409715 100644
--- a/src/armnnSerializer/ArmnnSchema.fbs
+++ b/src/armnnSerializer/ArmnnSchema.fbs
@@ -170,7 +170,8 @@
Fill = 57,
Rank = 58,
LogicalBinary = 59,
- Reduce = 60
+ Reduce = 60,
+ Cast = 61
}
// Base layer table to be used as part of other layers
@@ -219,6 +220,10 @@
axis:int;
}
+table CastLayer {
+ base:LayerBase;
+}
+
enum ComparisonOperation : byte {
Equal = 0,
Greater = 1,
@@ -964,7 +969,8 @@
FillLayer,
RankLayer,
LogicalBinaryLayer,
- ReduceLayer
+ ReduceLayer,
+ CastLayer
}
table AnyLayer {