aboutsummaryrefslogtreecommitdiff
path: root/src/backends/test/Conv2dTestImpl.hpp
diff options
context:
space:
mode:
authorJames Conroy <james.conroy@arm.com>2018-10-31 11:47:53 +0000
committerJames Conroy <james.conroy@arm.com>2018-10-31 12:06:53 +0000
commit45a9b775bf63283320315d90e4e9a6c641df6e20 (patch)
treee1f0d33d98410255a6804ea9cccf16805fc6080f /src/backends/test/Conv2dTestImpl.hpp
parentd84216a013445e86183e39c8b5b904836c71a95b (diff)
downloadarmnn-45a9b775bf63283320315d90e4e9a6c641df6e20.tar.gz
IVGCVSW-2102: Fix Pooling2D CpuRef indexing bug
* Fixes bug when calcuating indexes for NHWC in Pooling2D CpuRef implementation, it now uses TensorBufferArrayView. * Adds 2-Channel unit tests for Pooling2d on CpuRef, Cl and Neon. The single channel tests were not properly exercising Pooling2d using NHWC data layout. * Refactors Pooling2D NHWC tests so that the input and output data are permuted to NHWC when necessary, instead of hard coding the data in NHWC format. Change-Id: I5b9d41ed425ff283ea8c8ef6b1266ae0bc80f43b
Diffstat (limited to 'src/backends/test/Conv2dTestImpl.hpp')
-rwxr-xr-xsrc/backends/test/Conv2dTestImpl.hpp21
1 files changed, 0 insertions, 21 deletions
diff --git a/src/backends/test/Conv2dTestImpl.hpp b/src/backends/test/Conv2dTestImpl.hpp
index 7a3f452515..ce3e435d1a 100755
--- a/src/backends/test/Conv2dTestImpl.hpp
+++ b/src/backends/test/Conv2dTestImpl.hpp
@@ -62,27 +62,6 @@ void ApplyBias(std::vector<T>& v, float vScale, int32_t vOffset,
}
}
-template<typename T>
-armnn::TensorInfo GetTensorInfo(unsigned int numberOfBatches,
- unsigned int numberOfChannels,
- unsigned int height,
- unsigned int width,
- const armnn::DataLayoutIndexed& layout)
-{
- switch (layout.GetDataLayout())
- {
- case armnn::DataLayout::NCHW:
- return armnn::TensorInfo({numberOfBatches, numberOfChannels, height, width}, armnn::GetDataType<T>());
- case armnn::DataLayout ::NHWC:
- return armnn::TensorInfo({numberOfBatches, height, width, numberOfChannels}, armnn::GetDataType<T>());
- default:
- throw armnn::InvalidArgumentException("unknown data layout ["
- + std::to_string(static_cast<int>(layout.GetDataLayout())) + "]");
- }
-}
-
-
-
template<typename T, typename B>
LayerTestResult<T, 4> SimpleConvolution2dTestImpl(armnn::IWorkloadFactory& workloadFactory,
const boost::multi_array<T, 4>& originalInput,