24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_REDUCE_MEAN_H
25 #define ARM_COMPUTE_CL_REDUCE_MEAN_H
45 CLReduceMean(std::shared_ptr<IMemoryManager> memory_manager =
nullptr);
100 std::vector<CLReductionOperation> _reduction_kernels;
101 std::vector<CLTensor> _reduced_outs;
Base class for all functions.
Basic function to run opencl::ClDequantize that dequantizes an input tensor.
static Status validate(const ITensorInfo *input, const Coordinates &reduction_axis, bool keep_dims, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of CLReduceMean.
void configure(ICLTensor *input, const Coordinates &reduction_axis, bool keep_dims, ICLTensor *output)
Configure kernel.
Interface for OpenCL tensor.
Basic implementation of the OpenCL tensor interface.
Basic function to simulate a quantization layer.
void run() override
Run the kernels contained in the function.
CLReduceMean(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Default constructor.
Basic function to run opencl::kernels::ClReshapeKernel.
Basic function to perform reduce operation.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.