24 #ifndef ARM_COMPUTE_CLMEANSTDDEVKERNEL_H 25 #define ARM_COMPUTE_CLMEANSTDDEVKERNEL_H 61 void configure(
const ICLImage *
input,
float *mean, cl::Buffer *global_sum,
float *stddev =
nullptr, cl::Buffer *global_sum_squared =
nullptr);
71 void configure(
const CLCompileContext &compile_context,
const ICLImage *input,
float *mean, cl::Buffer *global_sum,
float *stddev =
nullptr, cl::Buffer *global_sum_squared =
nullptr);
82 static Status validate(
const ITensorInfo *input,
float *mean, cl::Buffer *global_sum,
float *stddev =
nullptr, cl::Buffer *global_sum_squared =
nullptr);
85 void run(
const Window &window, cl::CommandQueue &queue)
override;
93 cl::Buffer *_global_sum;
94 cl::Buffer *_global_sum_squared;
Affinity at buffer level.
Container for 2D border size.
ICLTensor ICLImage
Interface for OpenCL images.
Store the tensor's metadata.
Common interface for all the OpenCL kernels.
Copyright (c) 2017-2021 Arm Limited.
Interface for the kernel to calculate mean and standard deviation of input image pixels.
Interface for OpenCL tensor.
Describe a multidimensional execution window.
Status validate(const ITensorInfo *scores_in, const ITensorInfo *boxes_in, const ITensorInfo *batch_splits_in, const ITensorInfo *scores_out, const ITensorInfo *boxes_out, const ITensorInfo *classes, const ITensorInfo *batch_splits_out, const ITensorInfo *keeps, const ITensorInfo *keeps_size, const BoxNMSLimitInfo info)