21.02
|
Function to calculate the index of the minimum or maximum values in a tensor based on an axis. More...
#include <NEArgMinMaxLayer.h>
Public Member Functions | |
NEArgMinMaxLayer (std::shared_ptr< IMemoryManager > memory_manager=nullptr) | |
Constructor. More... | |
NEArgMinMaxLayer (const NEArgMinMaxLayer &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEArgMinMaxLayer & | operator= (const NEArgMinMaxLayer &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEArgMinMaxLayer (NEArgMinMaxLayer &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEArgMinMaxLayer & | operator= (NEArgMinMaxLayer &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEArgMinMaxLayer () | |
Default destructor. More... | |
void | configure (ITensor *input, int axis, ITensor *output, const ReductionOperation &op) |
Set the input and output tensors. 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, int axis, const ITensorInfo *output, const ReductionOperation &op) |
Static function to check if given info will lead to a valid configuration of NEArgMinMaxLayer. More... | |
Function to calculate the index of the minimum or maximum values in a tensor based on an axis.
This function calls the following Neon kernels:
Definition at line 50 of file NEArgMinMaxLayer.h.
NEArgMinMaxLayer | ( | std::shared_ptr< IMemoryManager > | memory_manager = nullptr | ) |
Constructor.
Definition at line 38 of file NEArgMinMaxLayer.cpp.
References ARM_COMPUTE_UNUSED.
|
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, |
int | axis, | ||
ITensor * | output, | ||
const ReductionOperation & | op | ||
) |
Set the input and output tensors.
[in] | input | Input source tensor. Data types supported: QASYMM8_SIGNED/QASYMM8/S32/F16/F32. |
[in] | axis | Axis to find max/min index. |
[out] | output | Output source tensor. Data types supported: U32/S32. |
[in] | op | Operation to perform: min or max |
Definition at line 43 of file NEArgMinMaxLayer.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)
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 54 of file NEArgMinMaxLayer.cpp.
|
static |
Static function to check if given info will lead to a valid configuration of NEArgMinMaxLayer.
[in] | input | Input source tensor info. Data types supported: QASYMM8_SIGNED/QASYMM8/S32/F16/F32. |
[in] | axis | Axis to find max/min index. |
[in] | output | Output source tensor info. Data types supported: U32/S32. |
[in] | op | Operation to perform: min or max |
Definition at line 48 of file NEArgMinMaxLayer.cpp.
References arm_compute::ARG_IDX_MAX, arm_compute::ARG_IDX_MIN, ARM_COMPUTE_RETURN_ERROR_ON_MSG, and NEReductionOperation::validate().
Referenced by arm_compute::test::validation::DATA_TEST_CASE().