diff options
| author | Kevin May <kevin.may@arm.com> | 2019-10-02 14:07:47 +0100 |
|---|---|---|
| committer | Kevin May <kevin.may@arm.com> | 2019-10-03 11:56:18 +0000 |
| commit | ce5045a00485f8a8c35814c0781ccbcca5678e5c (patch) | |
| tree | 7481fbdfd859f3edd24c1bf99830a0c89d6bb9ab /include/armnn/Descriptors.hpp | |
| parent | d47a064ab4c38559c6be931cb1771feb6e026ea4 (diff) | |
| download | armnn-ce5045a00485f8a8c35814c0781ccbcca5678e5c.tar.gz | |
IVGCVSW-3932 Add frontend for INSTANCE_NORMALIZATION
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: Ib152148ccd8d2733c617d0cf9402661fc6b71316
Diffstat (limited to 'include/armnn/Descriptors.hpp')
| -rw-r--r-- | include/armnn/Descriptors.hpp | 20 |
1 files changed, 20 insertions, 0 deletions
diff --git a/include/armnn/Descriptors.hpp b/include/armnn/Descriptors.hpp index c973089df6..5bf4043afa 100644 --- a/include/armnn/Descriptors.hpp +++ b/include/armnn/Descriptors.hpp @@ -468,6 +468,26 @@ struct BatchNormalizationDescriptor DataLayout m_DataLayout; }; +/// An InstanceNormalizationDescriptor for InstanceNormalizationLayer +struct InstanceNormalizationDescriptor +{ + InstanceNormalizationDescriptor() + : m_Gamma(1.0f) + , m_Beta(0.0f) + , m_Eps(1e-12f) + , m_DataLayout(DataLayout::NCHW) + {} + + /// Gamma, the scale scalar value applied for the normalized tensor. Defaults to 1.0. + float m_Gamma; + /// Beta, the offset scalar value applied for the normalized tensor. Defaults to 1.0. + float m_Beta; + /// Epsilon, small scalar value added to variance to avoid dividing by zero. Defaults to 1e-12f. + float m_Eps; + /// The data layout to be used (NCHW, NHWC). + DataLayout m_DataLayout; +}; + /// A BatchToSpaceNdDescriptor for the BatchToSpaceNdLayer. struct BatchToSpaceNdDescriptor { |