diff options
Diffstat (limited to 'src/backends/backendsCommon/test/SoftmaxTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/SoftmaxTestImpl.hpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp b/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp index 1e145a1a2e..2a239866f7 100644 --- a/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp +++ b/src/backends/backendsCommon/test/SoftmaxTestImpl.hpp @@ -62,8 +62,9 @@ LayerTestResult<T, 2> SimpleSoftmaxTestImpl(armnn::IWorkloadFactory& workloadFac outputHandle->Allocate(); CopyDataToITensorHandle(inputHandle.get(), &input[0][0]); - workloadFactory.Finalize(); + workloadFactory.Acquire(); workload->Execute(); + workloadFactory.Release(); CopyDataFromITensorHandle(&ret.output[0][0], outputHandle.get()); @@ -141,9 +142,10 @@ LayerTestResult<T, 2> CompareSoftmaxTestImpl(armnn::IWorkloadFactory& workloadFa CopyDataToITensorHandle(inputHandle.get(), &input[0][0]); CopyDataToITensorHandle(inputHandleRef.get(), &input[0][0]); - workloadFactory.Finalize(); + workloadFactory.Acquire(); workload->Execute(); - refWorkloadFactory.Finalize(); + workloadFactory.Release(); + workloadRef->Execute(); CopyDataFromITensorHandle(&ret.output[0][0], outputHandle.get()); |