From a121de3edee75180d335d8be42cdd6620a6063a5 Mon Sep 17 00:00:00 2001 From: nikraj01 Date: Wed, 29 May 2019 10:51:05 +0100 Subject: IVGCVSW-3168 Refactor reference softmax workload into a single workload Change-Id: Ie290efcbb9e3a6365cbd630cb2041e7b0f542505 Signed-off-by: nikraj01 --- src/backends/backendsCommon/test/WorkloadDataValidation.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/backends/backendsCommon/test/WorkloadDataValidation.cpp') diff --git a/src/backends/backendsCommon/test/WorkloadDataValidation.cpp b/src/backends/backendsCommon/test/WorkloadDataValidation.cpp index 94bef9b50a..795791fc5e 100644 --- a/src/backends/backendsCommon/test/WorkloadDataValidation.cpp +++ b/src/backends/backendsCommon/test/WorkloadDataValidation.cpp @@ -78,7 +78,7 @@ BOOST_AUTO_TEST_CASE(SoftmaxQueueDescriptor_Validate_WrongInputHeight) AddOutputToWorkload(invalidData, invalidInfo, outputTensorInfo, nullptr); //Invalid argument exception is expected, because height != 1. - BOOST_CHECK_THROW(RefSoftmaxFloat32Workload(invalidData, invalidInfo), armnn::InvalidArgumentException); + BOOST_CHECK_THROW(RefSoftmaxWorkload(invalidData, invalidInfo), armnn::InvalidArgumentException); } BOOST_AUTO_TEST_CASE(FullyConnectedQueueDescriptor_Validate_RequiredDataMissing) -- cgit v1.2.1