aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NECol2Im.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/NEON/functions/NECol2Im.cpp')
-rw-r--r--src/runtime/NEON/functions/NECol2Im.cpp7
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);
+}