aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp
diff options
context:
space:
mode:
authorKeith Davis <keith.davis@arm.com>2020-08-27 15:38:12 +0100
committerKeithARM <keith.davis@arm.com>2020-08-31 16:05:39 +0000
commit33a626f14c8db13967ba2a933388223d5bf733c3 (patch)
tree109319777c0f638be639d7d723a417abe97d843c /src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp
parent826a543547f325bfac7da507d19a55af9c138a50 (diff)
downloadarmnn-33a626f14c8db13967ba2a933388223d5bf733c3.tar.gz
IVGCVSW-5231 Remove CreateTensorHandle in the test where there is NO_DEPRECATE_WARN
* Done for all elementwise layers, Activation, BatchNorm, BatchToSpace Signed-off-by: Keith Davis <keith.davis@arm.com> Change-Id: Id1d15a0960233026aecf7a07e0d3f006e07e4abf
Diffstat (limited to 'src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp')
-rw-r--r--src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp9
1 files changed, 5 insertions, 4 deletions
diff --git a/src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp b/src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp
index 863e988c9a..5d37e934ea 100644
--- a/src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp
+++ b/src/backends/backendsCommon/test/layerTests/ElementwiseUnaryTestImpl.hpp
@@ -42,6 +42,7 @@ LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper(
int quantOffset,
const unsigned int outShape[NumDims],
std::vector<float> outValues,
+ const armnn::ITensorHandleFactory& tensorHandleFactory,
float outQuantScale,
int outQuantOffset)
{
@@ -58,10 +59,8 @@ LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper(
LayerTestResult<T, NumDims> ret(outputTensorInfo);
- ARMNN_NO_DEPRECATE_WARN_BEGIN
- std::unique_ptr<armnn::ITensorHandle> inputHandle = workloadFactory.CreateTensorHandle(inputTensorInfo);
- std::unique_ptr<armnn::ITensorHandle> outputHandle = workloadFactory.CreateTensorHandle(outputTensorInfo);
- ARMNN_NO_DEPRECATE_WARN_END
+ std::unique_ptr<armnn::ITensorHandle> inputHandle = tensorHandleFactory.CreateTensorHandle(inputTensorInfo);
+ std::unique_ptr<armnn::ITensorHandle> outputHandle = tensorHandleFactory.CreateTensorHandle(outputTensorInfo);
armnn::ElementwiseUnaryDescriptor desc(op);
armnn::ElementwiseUnaryQueueDescriptor qDesc;
@@ -97,6 +96,7 @@ LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper(
std::vector<float> values,
const unsigned int outShape[NumDims],
std::vector<float> outValues,
+ const armnn::ITensorHandleFactory& tensorHandleFactory,
float quantScale = 1.0f,
int quantOffset = 0)
{
@@ -110,6 +110,7 @@ LayerTestResult<T, NumDims> ElementwiseUnaryTestHelper(
quantOffset,
outShape,
outValues,
+ tensorHandleFactory,
quantScale,
quantOffset);
} \ No newline at end of file