Method configure

src/cpu/kernels/CpuActivationKernel.cpp:203–253

} // namespace

void CpuActivationKernel::configure(const ITensorInfo *src, ITensorInfo *dst, ActivationLayerInfo activation_info)
{
    ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuActivationKernel::configure");
    ARM_COMPUTE_ERROR_ON_NULLPTR(src);
    ARM_COMPUTE_ERROR_THROW_ON(CpuActivationKernel::validate(src, dst, activation_info));

    heuristics::CpuActivationKernelHeuristics heuristics(src, dst, activation_info);
    _heuristics = std::move(heuristics);

    _src_padding = src->padding();
    _inplace     = (dst == nullptr);
    if (!_inplace)
    {
        // dst auto inizialitation if not yet initialized
        auto_init_if_empty(*dst, *src->clone());
        _dst_padding = dst->padding();
    }

    const auto *uk = _heuristics.kernel();
    ARM_COMPUTE_ERROR_ON_NULLPTR(uk);

    _name = std::string("CpuActivationKernel").append("/").append(uk->name);

#ifdef __aarch64__
    // Initialise lut_manager
    LUTManager &lut_manager = LUTManager::get_instance();

    // TODO (COMPMID-7511): delegate to LUTManager
    if ((src->data_type() == DataType::QASYMM8 || src->data_type() == DataType::QASYMM8_SIGNED) &&
        activation_info.activation() != ActivationFunction::RELU)
    {
        ActivationLayerInfo::LookupTable256 tmp_lut;
        init_lut(activation_info.activation(), src->data_type(), src->quantization_info().uniform(),
                 (dst) ? dst->quantization_info().uniform() : src->quantization_info().uniform(), tmp_lut,
                 activation_info.a(), activation_info.b());
        activation_info.setLookupTable256(tmp_lut);
    }

    // Kernel specific logic should be mirrored in prepare()
    if (std::string(uk->name) == "sve_fp16_activation_lut")
    {
        // Create info using init list.
        const LUTInfo info = {activation_info.activation(), activation_info.a(), activation_info.b(), src->data_type(),
                              src->quantization_info().uniform()};
        activation_info.setLookupTable65536((lut_manager.get_lut_table<LookupTable65536>(info)));
    }
#endif // __aarch64__
    _act_info = activation_info;

    ICPPKernel::configure(_heuristics.window());
}

Status
CpuActivationKernel::validate(const ITensorInfo *src, const ITensorInfo *dst, const ActivationLayerInfo &act_info)