70 _kernel =
create_kernel(compile_context,
"accumulate_weighted");
74 _kernel.setArg(idx++, alpha);
92 _kernel =
create_kernel(compile_context,
"accumulate_squared");
96 _kernel.setArg(idx++, shift);
void configure(const ICLTensor *input, float alpha, ICLTensor *accum)
Set the input and accumulation images, and the scale value.
void configure(const ICLTensor *input, uint32_t shift, ICLTensor *accum)
Set the input and accumulation tensors and the shift value.
1 channel, 1 U8 per channel
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
static CLKernelLibrary & get()
Access the KernelLibrary singleton.
Copyright (c) 2017-2021 Arm Limited.
void configure(const ICLTensor *input, ICLTensor *output, unsigned int num_elems_processed_per_iteration, bool border_undefined=false, const BorderSize &border_size=BorderSize())
Configure the kernel.
cl::Kernel create_kernel(const CLCompileContext &ctx, const std::string &kernel_name, const std::set< std::string > &build_opts=std::set< std::string >())
Creates an opencl kernel using a compile context.
static constexpr unsigned int num_arguments_per_2D_tensor()
Returns the number of arguments enqueued per 2D tensor object.
1 channel, 1 S16 per channel
#define ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
Interface for OpenCL tensor.
Manages all the OpenCL kernels compilation and caching, provides accessors for the OpenCL Context...
Wrapper to configure the Khronos OpenCL C++ header.
unsigned int num_elems_processed_per_iteration
void configure(const ICLTensor *input, ICLTensor *accum)
Set the input and accumulation tensors.