aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp')
-rw-r--r--src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp11
1 files changed, 8 insertions, 3 deletions
diff --git a/src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp b/src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp
index 2f9c02dbb6..6d4a28db26 100644
--- a/src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp
+++ b/src/runtime/CL/functions/CLNonMaximaSuppression3x3.cpp
@@ -32,16 +32,21 @@ using namespace arm_compute;
void CLNonMaximaSuppression3x3::configure(ICLTensor *input, ICLTensor *output, BorderMode border_mode)
{
+ configure(CLKernelLibrary::get().get_compile_context(), input, output, border_mode);
+}
+
+void CLNonMaximaSuppression3x3::configure(const CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output, BorderMode border_mode)
+{
auto k = arm_compute::support::cpp14::make_unique<CLNonMaximaSuppression3x3Kernel>();
- k->configure(input, output, border_mode == BorderMode::UNDEFINED);
+ k->configure(compile_context, input, output, border_mode == BorderMode::UNDEFINED);
_kernel = std::move(k);
if(border_mode != BorderMode::UNDEFINED)
{
- _border_handler.configure(input, _kernel->border_size(), BorderMode::CONSTANT);
+ _border_handler.configure(compile_context, input, _kernel->border_size(), BorderMode::CONSTANT);
}
else
{
- _border_handler.configure(input, _kernel->border_size(), BorderMode::UNDEFINED);
+ _border_handler.configure(compile_context, input, _kernel->border_size(), BorderMode::UNDEFINED);
}
}