24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLREDUCTIONOPERATION_H
25 #define ARM_COMPUTE_CLREDUCTIONOPERATION_H
38 class CLCompileContext;
39 class CLReductionOperationKernel;
99 bool keep_dims =
true);
115 bool keep_dims =
true);
125 std::unique_ptr<CLReductionOperationKernel> _reduction_kernel;
127 unsigned int _reduction_axis;
128 bool _is_reshape_required;
CLReductionOperation(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default Constructor.
~CLReductionOperation()
Default Destructor.
Base class for all functions.
Interface for OpenCL tensor.
Basic implementation of the OpenCL tensor interface.
ReductionOperation
Available reduction operations.
void run() override
Run the kernels contained in the function.
Basic function to run opencl::kernels::ClReshapeKernel.
CLReductionOperation & operator=(const CLReductionOperation &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Copyright (c) 2017-2024 Arm Limited.
void configure(ICLTensor *input, ICLTensor *output, unsigned int axis, ReductionOperation op, bool keep_dims=true)
Set the input and output tensors.
Store the tensor's metadata.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, unsigned int axis, ReductionOperation op, bool keep_dims=true)
Static function to check if given info will lead to a valid configuration of CLReductionOperation.
Perform reduction operation.