diff options
Diffstat (limited to 'src/armnn/layers/BatchNormalizationLayer.cpp')
| -rw-r--r-- | src/armnn/layers/BatchNormalizationLayer.cpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/armnn/layers/BatchNormalizationLayer.cpp b/src/armnn/layers/BatchNormalizationLayer.cpp index 625e0d472d..ce351a4376 100644 --- a/src/armnn/layers/BatchNormalizationLayer.cpp +++ b/src/armnn/layers/BatchNormalizationLayer.cpp @@ -27,6 +27,7 @@ std::unique_ptr<IWorkload> BatchNormalizationLayer::CreateWorkload(const IWorklo ARMNN_ASSERT_MSG(m_Gamma != nullptr, "BatchNormalizationLayer: Gamma data should not be null."); BatchNormalizationQueueDescriptor descriptor; + SetAdditionalInfo(descriptor); descriptor.m_Mean = m_Mean.get(); descriptor.m_Variance = m_Variance.get(); |