aboutsummaryrefslogtreecommitdiff
path: root/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp')
-rw-r--r--src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp6
1 files changed, 6 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp
index ac2595b6bf..3e6223ab79 100644
--- a/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp
+++ b/src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp
@@ -49,6 +49,12 @@ LayerTestResult<T, 2> SimpleFullyConnectedTestImpl(
ExecuteWorkload(*workload, memoryManager);
+ if (workloadFactory.GetBackendId() == armnn::Compute::CpuRef)
+ {
+ workload->PostAllocationConfigure();
+ workload->Execute();
+ }
+
CopyDataFromITensorHandle(&result.output[0][0], outputHandle.get());
return result;