diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp b/src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp index 4433a2a4b2..021734aaa6 100644 --- a/src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp +++ b/src/armnn/backends/ClWorkloads/ClBatchNormalizationFloatWorkload.cpp @@ -9,6 +9,8 @@ #include "backends/ArmComputeTensorUtils.hpp" #include "backends/ClLayerSupport.hpp" +#include "ClWorkloadUtils.hpp" + namespace armnn { using namespace armcomputetensorutils; @@ -91,4 +93,4 @@ void ClBatchNormalizationFloatWorkload::FreeUnusedTensors() FreeTensorIfUnused(m_Beta); } -} //namespace armnn
\ No newline at end of file +} //namespace armnn |