39 auto k = std::make_unique<CLAccumulateKernel>();
40 k->configure(compile_context, input, accum);
41 _kernel = std::move(k);
51 auto k = std::make_unique<CLAccumulateWeightedKernel>();
52 k->configure(compile_context, input, alpha, accum);
53 _kernel = std::move(k);
63 auto k = std::make_unique<CLAccumulateSquaredKernel>();
64 k->configure(compile_context, input, shift, accum);
65 _kernel = std::move(k);
static CLKernelLibrary & get()
Access the KernelLibrary singleton.
Copyright (c) 2017-2021 Arm Limited.
void configure(const ICLTensor *input, ICLTensor *accum)
Set the input and accumulation tensors.
Interface for OpenCL tensor.
void configure(const ICLTensor *input, uint32_t shift, ICLTensor *accum)
Set the input and accumulation tensors and the shift value.
void configure(const ICLTensor *input, float alpha, ICLTensor *accum)
Set the input and accumulation tensors, and the scale value.