diff options
Diffstat (limited to 'src/backends/backendsCommon/test/BatchNormTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/BatchNormTestImpl.hpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/backendsCommon/test/BatchNormTestImpl.hpp b/src/backends/backendsCommon/test/BatchNormTestImpl.hpp index 6325130218..d63f0b5610 100644 --- a/src/backends/backendsCommon/test/BatchNormTestImpl.hpp +++ b/src/backends/backendsCommon/test/BatchNormTestImpl.hpp @@ -16,6 +16,8 @@ #include <test/TensorHelpers.hpp> +#include <DataLayoutIndexed.hpp> + template<typename T> LayerTestResult<T, 4> BatchNormTestImpl( armnn::IWorkloadFactory& workloadFactory, @@ -30,7 +32,7 @@ LayerTestResult<T, 4> BatchNormTestImpl( armnn::TensorInfo inputTensorInfo(inputOutputTensorShape, armnn::GetDataType<T>()); armnn::TensorInfo outputTensorInfo(inputOutputTensorShape, armnn::GetDataType<T>()); - armnn::DataLayoutIndexed dataLayoutIndexed(dataLayout); + armnnUtils::DataLayoutIndexed dataLayoutIndexed(dataLayout); armnn::TensorInfo tensorInfo({ inputOutputTensorShape[dataLayoutIndexed.GetChannelsIndex()] }, armnn::GetDataType<T>()); @@ -188,4 +190,4 @@ LayerTestResult<T,4> BatchNormTestNhwcImpl( CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); return ret; -}
\ No newline at end of file +} |