Method configure

src/gpu/cl/operators/ClScatter.cpp:66–101

}

void ClScatter::configure(const CLCompileContext &compile_context,
                          const ITensorInfo      *src,
                          const ITensorInfo      *updates,
                          const ITensorInfo      *indices,
                          ITensorInfo            *dst,
                          const ScatterInfo      &info)
{
    ARM_COMPUTE_ERROR_ON_NULLPTR(updates, indices, dst);
    ARM_COMPUTE_LOG_PARAMS(src, indices, dst, info);

    // Perform validation step
    ARM_COMPUTE_ERROR_THROW_ON(validate(src, updates, indices, dst, info));
    _fill_zero = info.zero_initialization;

    // If necessary, create fill kernel to fill dst tensor.
    if (_fill_zero)
    {
        auto f = std::make_unique<kernels::ClFillKernel>();
        f->configure(compile_context, dst, PixelValue(0.0f));
        _fill_kernel = std::move(f);
    }
    else if (src != dst) // Check whether copying is necessary
    {
        // Fill dst with src copy here.
        auto j = std::make_unique<kernels::ClCopyKernel>();
        j->configure(compile_context, src, dst);
        _copy_kernel = std::move(j);
        _run_copy    = true;
    }

    // Configure ClScatterKernel
    auto k = std::make_unique<kernels::ClScatterKernel>();
    k->set_target(CLScheduler::get().target());
    k->configure(compile_context, updates, indices, dst, info);
    _scatter_kernel = std::move(k);
}

void ClScatter::run(ITensorPack &tensors)
{