diff options
| author | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-10-26 19:05:32 +0100 |
|---|---|---|
| committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2018-11-08 12:00:31 +0000 |
| commit | 421405b6a21b124288a750e2da26dc01eb7391cb (patch) | |
| tree | 35f5655ce9d8b5921cb03630534f532e4eb47bf5 /src/runtime/NEON/functions/NEDeconvolutionLayer.cpp | |
| parent | f1adf11c776aebaa8da1b8644a4ba2453afd2b81 (diff) | |
| download | ComputeLibrary-421405b6a21b124288a750e2da26dc01eb7391cb.tar.gz | |
COMPMID-1675: Add SVE support
Change-Id: I86679adff556b6ffc9929b35cbf1b59b3958bdb1
Diffstat (limited to 'src/runtime/NEON/functions/NEDeconvolutionLayer.cpp')
| -rw-r--r-- | src/runtime/NEON/functions/NEDeconvolutionLayer.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp index c2d7d6a9d9..6887a0a8cd 100644 --- a/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp +++ b/src/runtime/NEON/functions/NEDeconvolutionLayer.cpp @@ -27,7 +27,7 @@ #include "arm_compute/core/Utils.h" #include "arm_compute/core/Validate.h" #include "arm_compute/core/utils/misc/ShapeCalculator.h" -#include "arm_compute/runtime/CPP/CPPScheduler.h" +#include "arm_compute/runtime/NEON/NEScheduler.h" using namespace arm_compute; using namespace arm_compute::misc::shape_calculator; @@ -166,7 +166,7 @@ void NEDeconvolutionLayer::prepare() // Run weights flipping and mark original weights tensor as unused _weights_flipped.allocator()->allocate(); - CPPScheduler::get().schedule(&_flip_weights, Window::DimZ); + NEScheduler::get().schedule(&_flip_weights, Window::DimZ); _original_weights->mark_as_unused(); // Prepare convolution |