24 #ifndef __ARM_COMPUTE_CLREDUCTIONOPERATION_H__ 25 #define __ARM_COMPUTE_CLREDUCTIONOPERATION_H__ 79 std::unique_ptr<CLTensor[]> _sums_vector{
nullptr };
80 std::unique_ptr<CLReductionOperationKernel[]> _reduction_kernels_vector{
nullptr };
81 std::unique_ptr<CLFillBorderKernel[]> _border_handlers_vector{
nullptr };
82 unsigned int _num_of_stages;
static Status validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op)
Static function to check if given info will lead to a valid configuration of CLReductionOperation.
Base class for all functions.
ReductionOperation
Available reduction operations.
void configure(ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op)
Set the input and output tensors.
Store the tensor's metadata.
This file contains all available output stages for GEMMLowp on OpenCL.
void run() override
Run the kernels contained in the function.
CLReductionOperation(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default Constructor.
Interface for OpenCL tensor.
Perform reduction operation.