24.02.1
|
Basic function to unpack a rank-R tensor into rank-(R-1) tensors. More...
#include <NEUnstack.h>
Public Member Functions | |
NEUnstack () | |
Default constructor. More... | |
NEUnstack (const NEUnstack &)=delete | |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEUnstack & | operator= (const NEUnstack &)=delete |
Prevent instances of this class from being copied (As this class contains pointers) More... | |
NEUnstack (NEUnstack &&)=delete | |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
NEUnstack & | operator= (NEUnstack &&)=delete |
Prevent instances of this class from being moved (As this class contains non movable objects) More... | |
~NEUnstack ()=default | |
Default destructor. More... | |
void | configure (const ITensor *input, const std::vector< ITensor * > &output_vector, int axis) |
Set the input, output and unstacking axis. 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 std::vector< ITensorInfo * > &output_vector, int axis) |
Static function to check if given info will lead to a valid configuration of NEUnstack. More... | |
Basic function to unpack a rank-R tensor into rank-(R-1) tensors.
This function calls the following functions:
Definition at line 42 of file NEUnstack.h.
NEUnstack | ( | ) |
Default constructor.
Definition at line 60 of file NEUnstack.cpp.
Prevent instances of this class from being copied (As this class contains pointers)
Prevent instances of this class from being moved (As this class contains non movable objects)
|
default |
Default destructor.
Set the input, output and unstacking axis.
Valid data layouts:
Valid data type configurations:
src | dst |
---|---|
All | All |
[in] | input | A tensor to be unstacked. Data type supported: All. |
[in,out] | output_vector | A vector of tensors. Data types supported: same as input . Note: The number of elements of the vector will be used as the number of slices to be taken from the axis. |
[in] | axis | The axis to unstack along. Valid values are [-R,R) where R is the input's rank. Negative values wrap around. |
Definition at line 65 of file NEUnstack.cpp.
References ARM_COMPUTE_ERROR_ON_NULLPTR, ARM_COMPUTE_ERROR_THROW_ON, ARM_COMPUTE_LOG_PARAMS, arm_compute::test::validation::input, Dimensions< T >::set(), arm_compute::test::validation::reference::slice(), tf_frozen_model_extractor::t, and NEUnstack::validate().
Prevent instances of this class from being copied (As this class contains pointers)
Prevent instances of this class from being moved (As this class contains non movable objects)
|
overridevirtual |
Run the kernels contained in the function.
For CPU kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 120 of file NEUnstack.cpp.
|
static |
Static function to check if given info will lead to a valid configuration of NEUnstack.
[in] | input | Input tensor info. Data type supported: All. |
[in] | output_vector | Vector of output tensors' info. Data types supported: same as input . |
[in] | axis | The axis to unstack along. Valid values are [-R,R) where R is the input's rank. Negative values wrap around. |
Definition at line 96 of file NEUnstack.cpp.
References ARM_COMPUTE_RETURN_ERROR_ON, ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR, ARM_COMPUTE_RETURN_ON_ERROR, arm_compute::test::validation::input, Dimensions< T >::set(), and NEStridedSlice::validate().
Referenced by NEUnstack::configure().