diff options
Diffstat (limited to 'src/backends/reference/workloads/BatchNormImpl.hpp')
| -rw-r--r-- | src/backends/reference/workloads/BatchNormImpl.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/reference/workloads/BatchNormImpl.hpp b/src/backends/reference/workloads/BatchNormImpl.hpp index fbcb2fdf5a..799e7a327b 100644 --- a/src/backends/reference/workloads/BatchNormImpl.hpp +++ b/src/backends/reference/workloads/BatchNormImpl.hpp @@ -10,6 +10,8 @@ #include <armnn/Tensor.hpp> +#include <DataLayoutIndexed.hpp> + #include <cmath> namespace armnn @@ -34,7 +36,7 @@ static void BatchNormImpl(NormData data, outputData, data.m_Parameters.m_DataLayout); - DataLayoutIndexed dataLayout(data.m_Parameters.m_DataLayout); + armnnUtils::DataLayoutIndexed dataLayout(data.m_Parameters.m_DataLayout); for (unsigned int c = 0; c < inputInfo.GetShape()[dataLayout.GetChannelsIndex()]; c++) { |