21.02
|
Basic function to perform reduce operation. More...
#include <NEReduceMean.h>
Public Member Functions | |
NEReduceMean (std::shared_ptr< IMemoryManager > memory_manager=nullptr) | |
Constructor. More... | |
NEReduceMean (const NEReduceMean &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEReduceMean & | operator= (const NEReduceMean &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEReduceMean (NEReduceMean &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEReduceMean & | operator= (NEReduceMean &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEReduceMean () | |
Default destructor. More... | |
void | configure (ITensor *input, const Coordinates &reduction_axis, bool keep_dims, ITensor *output) |
Configure kernel. More... | |
void | run () override |
Run the kernels contained in the function. More... | |
Public Member Functions inherited from IFunction | |
virtual | ~IFunction ()=default |
Destructor. More... | |
virtual void | prepare () |
Prepare the function for executing. More... | |
Static Public Member Functions | |
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. More... | |
Basic function to perform reduce operation.
Definition at line 40 of file NEReduceMean.h.
NEReduceMean | ( | std::shared_ptr< IMemoryManager > | memory_manager = nullptr | ) |
Constructor.
Definition at line 102 of file NEReduceMean.cpp.
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
delete |
Prevent instances of this class from being moved (As this class contains non movable objects)
|
default |
Default destructor.
void configure | ( | ITensor * | input, |
const Coordinates & | reduction_axis, | ||
bool | keep_dims, | ||
ITensor * | output | ||
) |
Configure kernel.
[in] | input | Source tensor. Data type supported: QASYMM8_SIGNED/QASYMM8/F16/F32 |
[in] | reduction_axis | Reduction axis vector. |
[in] | keep_dims | If positive, retains reduced dimensions with length 1. |
[out] | output | Destination tensor. Data type supported: Same as input |
Definition at line 113 of file NEReduceMean.cpp.
References TensorAllocator::allocate(), Tensor::allocator(), ARM_COMPUTE_ERROR_THROW_ON, arm_compute::auto_init_if_empty(), Dimensions< T >::begin(), arm_compute::misc::shape_calculator::calculate_reduce_mean_shape(), ICloneable< T >::clone(), NEDequantizationLayer::configure(), NEQuantizationLayer::configure(), NEReshapeLayer::configure(), arm_compute::convert_negative_axis(), ITensorInfo::data_type(), arm_compute::F32, ITensor::info(), Tensor::info(), arm_compute::test::validation::info, arm_compute::test::validation::input, arm_compute::is_data_type_quantized(), MemoryGroup::manage(), arm_compute::MEAN_SUM, ITensorInfo::num_channels(), Dimensions< T >::num_dimensions(), ITensorInfo::num_dimensions(), arm_compute::test::validation::output_shape, ITensorInfo::quantization_info(), TensorShape::remove_dimension(), TensorShape::set(), TensorInfo::set_data_type(), ITensorInfo::tensor_shape(), and NEReduceMean::validate().
|
delete |
Prevent instances of this class from being copied (As this class contains pointers)
|
delete |
Prevent instances of this class from being moved (As this class contains non movable objects)
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 194 of file NEReduceMean.cpp.
References INESimpleFunctionNoBorder::run(), and NEReshapeLayer::run().
|
static |
Static function to check if given info will lead to a valid configuration of NEReduceMean.
[in] | input | Source tensor. Data type supported: QASYMM8_SIGNED/QASYMM8/F16/F32 |
[in] | reduction_axis | Reduction axis vector. |
[in] | keep_dims | If positive, retains reduced dimensions with length 1. |
[in] | output | Destination tensor. Data type supported: Same as input |
Definition at line 108 of file NEReduceMean.cpp.
Referenced by NEReduceMean::configure().