aboutsummaryrefslogtreecommitdiff
path: root/src/armnnSerializer/SerializerUtils.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnSerializer/SerializerUtils.cpp')
-rw-r--r--src/armnnSerializer/SerializerUtils.cpp16
1 files changed, 16 insertions, 0 deletions
diff --git a/src/armnnSerializer/SerializerUtils.cpp b/src/armnnSerializer/SerializerUtils.cpp
index 5566abf7e4..045d6aac5c 100644
--- a/src/armnnSerializer/SerializerUtils.cpp
+++ b/src/armnnSerializer/SerializerUtils.cpp
@@ -28,6 +28,20 @@ armnnSerializer::ComparisonOperation GetFlatBufferComparisonOperation(armnn::Com
}
}
+armnnSerializer::LogicalBinaryOperation GetFlatBufferLogicalBinaryOperation(
+ armnn::LogicalBinaryOperation logicalBinaryOperation)
+{
+ switch (logicalBinaryOperation)
+ {
+ case armnn::LogicalBinaryOperation::LogicalAnd:
+ return armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalAnd;
+ case armnn::LogicalBinaryOperation::LogicalOr:
+ return armnnSerializer::LogicalBinaryOperation::LogicalBinaryOperation_LogicalOr;
+ default:
+ throw armnn::InvalidArgumentException("Logical Binary operation unknown");
+ }
+}
+
armnnSerializer::ConstTensorData GetFlatBufferConstTensorData(armnn::DataType dataType)
{
switch (dataType)
@@ -98,6 +112,8 @@ armnnSerializer::UnaryOperation GetFlatBufferUnaryOperation(armnn::UnaryOperatio
return armnnSerializer::UnaryOperation::UnaryOperation_Exp;
case armnn::UnaryOperation::Neg:
return armnnSerializer::UnaryOperation::UnaryOperation_Neg;
+ case armnn::UnaryOperation::LogicalNot:
+ return armnnSerializer::UnaryOperation::UnaryOperation_LogicalNot;
default:
throw armnn::InvalidArgumentException("Unary operation unknown");
}