From 495852f2adef1d11fbf13ce6347cf61973ce1a65 Mon Sep 17 00:00:00 2001 From: Ruomei Yan Date: Thu, 23 May 2019 11:37:33 +0100 Subject: IVGCVSW-3073 Refactor reference DepthwiseConvolution2d workloads Change-Id: I3cf8d9dbc4f8c95e0e2311505dd7e9f9069f1ab5 Signed-off-by: Ruomei Yan --- src/backends/backendsCommon/test/Conv2dTestImpl.hpp | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) (limited to 'src/backends/backendsCommon/test/Conv2dTestImpl.hpp') 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 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 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 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 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 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()); -- cgit v1.2.1