24 #ifndef ARM_COMPUTE_CPPTOPKV_H 25 #define ARM_COMPUTE_CPPTOPKV_H Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
Basic interface for functions which have a single CPP kernel.
void configure(const ITensor *predictions, const ITensor *targets, ITensor *output, const unsigned int k)
Set the input and output of the kernel.
Basic function to run 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.