diff options
Diffstat (limited to 'src/armnn/SerializeLayerParameters.hpp')
-rw-r--r-- | src/armnn/SerializeLayerParameters.hpp | 12 |
1 files changed, 11 insertions, 1 deletions
diff --git a/src/armnn/SerializeLayerParameters.hpp b/src/armnn/SerializeLayerParameters.hpp index d2c7d76d7d..b786c53882 100644 --- a/src/armnn/SerializeLayerParameters.hpp +++ b/src/armnn/SerializeLayerParameters.hpp @@ -1,5 +1,5 @@ // -// Copyright © 2017 Arm Ltd. All rights reserved. +// Copyright © 2017,2022 Arm Ltd. All rights reserved. // SPDX-License-Identifier: MIT // #pragma once @@ -35,6 +35,11 @@ template <> struct StringifyLayerParameters<BatchNormalizationDescriptor> static void Serialize(ParameterStringifyFunction& fn, const BatchNormalizationDescriptor& desc); }; +template <> struct StringifyLayerParameters<BatchMatMulDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const BatchMatMulDescriptor& desc); +}; + template <> struct StringifyLayerParameters<BatchToSpaceNdDescriptor> { static void Serialize(ParameterStringifyFunction& fn, const BatchToSpaceNdDescriptor& desc); @@ -85,6 +90,11 @@ template <> struct StringifyLayerParameters<FullyConnectedDescriptor> static void Serialize(ParameterStringifyFunction& fn, const FullyConnectedDescriptor& desc); }; +template <> struct StringifyLayerParameters<GatherDescriptor> +{ + static void Serialize(ParameterStringifyFunction& fn, const GatherDescriptor& desc); +}; + template <> struct StringifyLayerParameters<L2NormalizationDescriptor> { static void Serialize(ParameterStringifyFunction& fn, const L2NormalizationDescriptor& desc); |