24 #ifndef ARM_COMPUTE_NEON_REDUCE_MEAN_H 25 #define ARM_COMPUTE_NEON_REDUCE_MEAN_H 44 NEReduceMean(std::shared_ptr<IMemoryManager> memory_manager =
nullptr);
82 std::vector<NEReductionOperation> _reduction_kernels;
83 std::vector<Tensor> _reduced_outs;
Base class for all functions.
Store the tensor's metadata.
Basic function to simulate a quantization layer.
~NEReduceMean()
Default destructor.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
void configure(ITensor *input, const Coordinates &reduction_axis, bool keep_dims, ITensor *output)
Configure kernel.
Basic implementation of the tensor interface.
Basic function to run NEDequantizationLayerKernel that dequantizes an input tensor.
NEReduceMean(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
Basic function to run cpu::kernels::CpuReshapeKernel.
Basic function to perform reduce operation.
NEReduceMean & operator=(const NEReduceMean &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
void run() override
Run the kernels contained in the function.
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 NEReduceMean.