diff options
author | Bruno Goncalves <bruno.slackware@gmail.com> | 2019-04-26 21:03:24 -0300 |
---|---|---|
committer | Matteo Martincigh <matteo.martincigh@arm.com> | 2019-05-14 07:36:33 +0000 |
commit | 22972f04d6aa5c4d1269ed3be8bc5fb7b508dcda (patch) | |
tree | be61df59d9eb215bf1396a5a03797942066c6ba6 /src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp | |
parent | acad04e3cb6abcdd9a3fcf4584db1cbedb52cb47 (diff) | |
download | armnn-22972f04d6aa5c4d1269ed3be8bc5fb7b508dcda.tar.gz |
MLCE-101 Add dilation support for DepthWiseConv workload
Adds unit tests for dilated depthwise conv
Change-Id: Iad0a1b33d07fb0ef8f9f6edf0fd0f83a5800a36d
Signed-off-by: Bruno Goncalves <bruno.slackware@gmail.com>
Signed-off-by: Matthew Bentham <matthew.bentham@arm.com>
Diffstat (limited to 'src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp')
-rw-r--r-- | src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp b/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp index 0b917fc530..d15b48571a 100644 --- a/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp +++ b/src/backends/neon/workloads/NeonDepthwiseConvolutionWorkload.cpp @@ -117,7 +117,7 @@ NeonDepthwiseConvolutionWorkload::NeonDepthwiseConvolutionWorkload( input.info()->set_data_layout(aclDataLayout); output.info()->set_data_layout(aclDataLayout); - // Get the depth multiplier + // Get the depth multiplier const unsigned int depthMultiplier = weightInfo.GetShape()[0]; // Check for optimisation opportunities. |