aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CPP/functions/CPPTopKV.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/runtime/CPP/functions/CPPTopKV.cpp')
-rw-r--r--src/runtime/CPP/functions/CPPTopKV.cpp11
1 files changed, 9 insertions, 2 deletions
diff --git a/src/runtime/CPP/functions/CPPTopKV.cpp b/src/runtime/CPP/functions/CPPTopKV.cpp
index 2547e56a1d..3d64def804 100644
--- a/src/runtime/CPP/functions/CPPTopKV.cpp
+++ b/src/runtime/CPP/functions/CPPTopKV.cpp
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2019-2020 Arm Limited.
+ * Copyright (c) 2019-2021 Arm Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -25,16 +25,23 @@
#include "arm_compute/core/CPP/kernels/CPPTopKVKernel.h"
+#include "src/common/utils/Log.h"
+
namespace arm_compute
{
void CPPTopKV::configure(const ITensor *predictions, const ITensor *targets, ITensor *output, const unsigned int k)
{
+ ARM_COMPUTE_LOG_PARAMS(predictions, targets, output, k);
+
auto kernel = std::make_unique<CPPTopKVKernel>();
kernel->configure(predictions, targets, output, k);
_kernel = std::move(kernel);
}
-Status CPPTopKV::validate(const ITensorInfo *predictions, const ITensorInfo *targets, ITensorInfo *output, const unsigned int k)
+Status CPPTopKV::validate(const ITensorInfo *predictions,
+ const ITensorInfo *targets,
+ ITensorInfo *output,
+ const unsigned int k)
{
return CPPTopKVKernel::validate(predictions, targets, output, k);
}