diff options
Diffstat (limited to 'src/runtime/NEON/functions/NECol2Im.cpp')
-rw-r--r-- | src/runtime/NEON/functions/NECol2Im.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NECol2Im.cpp b/src/runtime/NEON/functions/NECol2Im.cpp index 2dc4ebd49a..2a923f3730 100644 --- a/src/runtime/NEON/functions/NECol2Im.cpp +++ b/src/runtime/NEON/functions/NECol2Im.cpp @@ -28,9 +28,14 @@ using namespace arm_compute; -void NECol2Im::configure(const ITensor *input, ITensor *output, std::pair<unsigned int, unsigned int> convolved_dims) +void NECol2Im::configure(const ITensor *input, ITensor *output, const Size2D &convolved_dims) { auto k = arm_compute::support::cpp14::make_unique<NECol2ImKernel>(); k->configure(input, output, convolved_dims); _kernel = std::move(k); } + +Error NECol2Im::validate(const ITensorInfo *input, const ITensorInfo *output, const Size2D &convolved_dims) +{ + return NECol2ImKernel::validate(input, output, convolved_dims); +} |