Compute Library
18.05
|
Basic function to run NEIm2ColKernel. More...
#include <NEIm2Col.h>
Public Member Functions | |
NEIm2Col () | |
Default constructor. More... | |
void | configure (const ITensor *input, ITensor *output, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, bool is_fully_connected=false, bool is_flatten=false) |
Configure the im2col NEON 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 ITensorInfo *output, const Size2D &kernel_dims, const PadStrideInfo &conv_info, bool has_bias, bool is_fully_connected, bool is_flatten) |
Static function to check if given info will lead to a valid configuration of NEIm2Col. More... | |
Basic function to run NEIm2ColKernel.
Definition at line 38 of file NEIm2Col.h.
NEIm2Col | ( | ) |
Default constructor.
void configure | ( | const ITensor * | input, |
ITensor * | output, | ||
const Size2D & | kernel_dims, | ||
const PadStrideInfo & | conv_info, | ||
bool | has_bias, | ||
bool | is_fully_connected = false , |
||
bool | is_flatten = false |
||
) |
Configure the im2col NEON kernel.
[in] | input | The input tensor to convert. 3 lower dimensions represent a single input [width, height, IFM], while every optional dimension from 4 and above represent a batch of inputs. Data types supported: QS8/QS16/QASYMM8/F16/F32 Note: QASYMM8 works only for has_bias = false |
[out] | output | The output tensor. Data types supported: Same as input |
[in] | kernel_dims | The kernel dimensions (width and height). |
[in] | conv_info | Contains padding and stride information described in PadStrideInfo. |
[in] | has_bias | In case biases are provided expands the matrix with 1. |
[in] | is_fully_connected | (Optional) Determines whether this function will be called by NEFullyConnectedLayer in order to validate the arguments |
[in] | is_flatten | (Optional) Determines whether this function will be called by NEFlattenLayer in order to validate the arguments |
|
overridevirtual |
Run the kernels contained in the function.
For NEON kernels:
For OpenCL kernels:
Implements IFunction.
|
static |
Static function to check if given info will lead to a valid configuration of NEIm2Col.
[in] | input | The input tensor to convert. 3 lower dimensions represent a single input [width, height, IFM], while every optional dimension from 4 and above represent a batch of inputs. Data types supported: QS8/QS16/QASYMM8/F16/F32 Note: QASYMM8 works only for has_bias = false |
[in] | output | The output tensor. Data types supported: Same as input |
[in] | kernel_dims | The kernel dimensions (width and height). |
[in] | conv_info | Contains padding and stride information described in PadStrideInfo. |
[in] | has_bias | In case biases are provided expands the matrix with 1. |
[in] | is_fully_connected | Determines whether this function will be called by NEFullyConnectedLayer in order to validate the arguments |
[in] | is_flatten | Determines whether this function will be called by NEFlattenLayer in order to validate the arguments |
Referenced by arm_compute::test::validation::DATA_TEST_CASE().