Compute Library
18.05
|
Basic function to run CPPUpsample. More...
#include <CPPUpsample.h>
Public Member Functions | |
void | configure (const ITensor *input, ITensor *output, const PadStrideInfo &info, unsigned int inner_border_right, unsigned int inner_border_top) |
Configure the upsample CPP kernel. More... | |
Public Member Functions inherited from ICPPSimpleFunction | |
ICPPSimpleFunction () | |
Constructor. More... | |
void | run () overridefinal |
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... | |
Basic function to run CPPUpsample.
Definition at line 36 of file CPPUpsample.h.
void configure | ( | const ITensor * | input, |
ITensor * | output, | ||
const PadStrideInfo & | info, | ||
unsigned int | inner_border_right, | ||
unsigned int | inner_border_top | ||
) |
Configure the upsample CPP kernel.
[in] | input | The input tensor to upsample. Data types supported: F32 |
[out] | output | The output tensor. Data types supported: Same as input |
[in] | info | Padding information |
[in] | inner_border_right | The number of zeros added to right edge of the input. |
[in] | inner_border_top | The number of zeros added to top edge of the input. |