24.02.1
|
Go to the documentation of this file.
36 auto kernel = std::make_unique<CPPTopKVKernel>();
37 kernel->configure(predictions, targets, output, k);
38 _kernel = std::move(kernel);
Interface for CPU tensor.
static Status validate(const ITensorInfo *predictions, const ITensorInfo *targets, ITensorInfo *output, const unsigned int k)
Static function to check if given info will lead to a valid configuration of CPPTopKVKernel.
static Status validate(const ITensorInfo *predictions, const ITensorInfo *targets, ITensorInfo *output, const unsigned int k)
Static function to check if given info will lead to a valid configuration of CPPTopKVKernel.
void configure(const ITensor *predictions, const ITensor *targets, ITensor *output, const unsigned int k)
Set the input and output of the kernel.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)