diff options
Diffstat (limited to 'src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp')
-rw-r--r-- | src/backends/backendsCommon/test/FullyConnectedTestImpl.hpp | 6 |
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; |