diff options
| author | Cathal Corbett <cathal.corbett@arm.com> | 2024-06-21 18:12:48 +0100 |
|---|---|---|
| committer | Cathal Corbett <cathal.corbett@arm.com> | 2024-06-27 14:02:33 +0000 |
| commit | b0cb2686607478339c29aa9cef7a58cf010ae59a (patch) | |
| tree | 6bd47ea9076a662c0771dd18f8a0a854bc106ad6 /src/backends/backendsCommon/test | |
| parent | 3604256857b1ea72972bc23c56b0254beeae865d (diff) | |
| download | armnn-b0cb2686607478339c29aa9cef7a58cf010ae59a.tar.gz | |
IVGCVSW-8445 Implement DepthwiseConvolution2d in TosaRef
Signed-off-by: Cathal Corbett <cathal.corbett@arm.com>
Change-Id: Ib733b9fe1a5b6f47dc110c7bb96cbea9d656df0d
Diffstat (limited to 'src/backends/backendsCommon/test')
| -rw-r--r-- | src/backends/backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp b/src/backends/backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp index a2c369b692..778b4823c3 100644 --- a/src/backends/backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp +++ b/src/backends/backendsCommon/test/DepthwiseConvolution2dEndToEndTests.hpp @@ -45,7 +45,7 @@ armnn::INetworkPtr CreateDepthwiseConvolution2dNetwork(const armnn::DepthwiseCon } // anonymous namespace -template<armnn::DataType ArmnnType, armnn::DataType ArmnnBType> +template<armnn::DataType ArmnnType, armnn::DataType ArmnnBType = ArmnnType> void DepthwiseConvolution2dEndToEnd(const std::vector<armnn::BackendId>& backends, armnn::DataLayout dataLayout) { @@ -168,6 +168,7 @@ void DepthwiseConvolution2dEndToEnd(const std::vector<armnn::BackendId>& backend { PermuteTensorNhwcToNchw(inputInfo, inputData); PermuteTensorNhwcToNchw(outputInfo, expectedOutputData); + PermuteTensorNhwcToNchw(weightsInfo, weightsData); } // Quantize data |