diff options
author | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 11:37:33 +0100 |
---|---|---|
committer | Ruomei Yan <ruomei.yan@arm.com> | 2019-05-23 11:51:39 +0100 |
commit | 495852f2adef1d11fbf13ce6347cf61973ce1a65 (patch) | |
tree | 9c26ac47362548e001ad7e5203212d2ef3226e9d /src/backends/backendsCommon/test | |
parent | 9b3983299f882c8d84c5abd0d40ca75a801ea7f2 (diff) | |
download | armnn-495852f2adef1d11fbf13ce6347cf61973ce1a65.tar.gz |
IVGCVSW-3073 Refactor reference DepthwiseConvolution2d workloads
Change-Id: I3cf8d9dbc4f8c95e0e2311505dd7e9f9069f1ab5
Signed-off-by: Ruomei Yan <ruomei.yan@arm.com>
Diffstat (limited to 'src/backends/backendsCommon/test')
-rw-r--r-- | src/backends/backendsCommon/test/Conv2dTestImpl.hpp | 11 |
1 files changed, 6 insertions, 5 deletions
diff --git a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp index 0533c77d53..73b170e930 100644 --- a/src/backends/backendsCommon/test/Conv2dTestImpl.hpp +++ b/src/backends/backendsCommon/test/Conv2dTestImpl.hpp @@ -456,7 +456,7 @@ LayerTestResult<T, 4> DepthwiseConvolution2dAsymmetricTestImpl( CopyDataToITensorHandle(inputHandle.get(), &batchedInput[0][0][0][0]); - workload->Execute(); + ExecuteWorkload(*workload, memoryManager); CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); @@ -601,7 +601,7 @@ LayerTestResult<T, 4> DepthwiseConvolution2dDepthMul1TestImpl( CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]); - workload->Execute(); + ExecuteWorkload(*workload, memoryManager); CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); @@ -804,7 +804,7 @@ LayerTestResult<T, 4> DepthwiseConvolution2dTestImpl( CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]); - workload->Execute(); + ExecuteWorkload(*workload, memoryManager); CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); @@ -910,7 +910,7 @@ LayerTestResult<T, 4> DepthwiseConvolution2dNhwcTestImpl( CopyDataToITensorHandle(inputHandle.get(), &batchedInput[0][0][0][0]); - workload->Execute(); + ExecuteWorkload(*workload, memoryManager); CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); @@ -1246,8 +1246,9 @@ LayerTestResult<T, 4> CompareDepthwiseConvolution2dTestImpl( CopyDataToITensorHandle(inputHandle.get(), &input[0][0][0][0]); CopyDataToITensorHandle(inputHandleRef.get(), &input[0][0][0][0]); - workload->Execute(); + ExecuteWorkload(*workload, memoryManager); + workloadRef->PostAllocationConfigure(); workloadRef->Execute(); CopyDataFromITensorHandle(&ret.output[0][0][0][0], outputHandle.get()); |