aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/SerializeLayerParameters.hpp
diff options
context:
space:
mode:
authorMike Kelly <mike.kelly@arm.com>2023-07-18 12:03:41 +0100
committermike.kelly <mike.kelly@arm.com>2023-07-21 16:01:50 +0000
commitb6de7a1444c09c0eb44c84a923c45c041b1f6092 (patch)
treec1d9e9241854e7d9bbb09da4f4a6f4652ff5c66c /src/armnn/SerializeLayerParameters.hpp
parentc32adef195b523854144737ca05180235f5ca824 (diff)
downloadarmnn-b6de7a1444c09c0eb44c84a923c45c041b1f6092.tar.gz
IVGCVSW-7830 Clean up
* Follow up review to clean up whitespace and copyright errors mentioned in https://review.mlplatform.org/c/ml/armnn/+/9885 * Added BinaryElementwiseOperation to .dot files * Refactored ConnectedToSplitterWithMoreThan4Dims function to more generally useful ConnectedToLayerType function Signed-off-by: Mike Kelly <mike.kelly@arm.com> Change-Id: I0e3d0895888f3a3f0a9758ce30bc031aba50812b
Diffstat (limited to 'src/armnn/SerializeLayerParameters.hpp')
-rw-r--r--src/armnn/SerializeLayerParameters.hpp5
1 files changed, 5 insertions, 0 deletions
diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp
index b786c53882..b90f9d85e8 100644
--- a/src/armnn/SerializeLayerParameters.hpp
+++ b/src/armnn/SerializeLayerParameters.hpp
@@ -75,6 +75,11 @@ template <> struct StringifyLayerParameters<DepthwiseConvolution2dDescriptor>
static void Serialize(ParameterStringifyFunction& fn, const DepthwiseConvolution2dDescriptor& desc);
};
+template <> struct StringifyLayerParameters<ElementwiseBinaryDescriptor>
+{
+ static void Serialize(ParameterStringifyFunction& fn, const ElementwiseBinaryDescriptor& desc);
+};
+
template <> struct StringifyLayerParameters<ElementwiseUnaryDescriptor>
{
static void Serialize(ParameterStringifyFunction& fn, const ElementwiseUnaryDescriptor& desc);