Method configure

src/cpu/kernels/CpuTopKVKernel.cpp:143–163

} // namespace

void CpuTopKVKernel::configure(const ITensorInfo *predictions, const ITensorInfo *targets, ITensorInfo *dst, uint32_t k)
{
    ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuTopKVKernel::configure");
    ARM_COMPUTE_UNUSED(targets); // workaround for a compiler bug about a false positive -Wunused-parameter
    ARM_COMPUTE_ERROR_ON_NULLPTR(predictions, targets, dst);
    ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(*predictions, *targets, *dst, k));

    const auto uk = CpuTopKVKernel::get_implementation<CpuTopKVKernelDataTypeISASelectorData>(
        CpuTopKVKernelDataTypeISASelectorData{predictions->data_type(), CPUInfo::get().get_isa()});

    ARM_COMPUTE_ERROR_ON_NULLPTR(uk);

    _run_method = uk->ukernel;
    _name       = std::string("CpuTopKVKernel").append("/").append(uk->name);
    _k          = k;
    // Auto initialize dst if not initialized
    auto_init_if_empty(*dst, TensorShape(predictions->dimension(1)), 1U, DataType::U8);
    // Configure kernel window
    Window win = calculate_max_window(*dst, Steps());
    ICpuKernel::configure(win);
}
size_t CpuTopKVKernel::get_mws(const CPUInfo &platform, size_t thread_count) const
{
    ARM_COMPUTE_UNUSED(thread_count);