24.02.1
|
Basic function to run NEFillBorderKernel. More...
#include <NEFillBorder.h>
Public Member Functions | |
NEFillBorder () | |
void | configure (ITensor *input, unsigned int border_width, BorderMode border_mode, const PixelValue &constant_border_value=PixelValue()) |
Initialize the function's source, destination and border_mode. 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... | |
Basic function to run NEFillBorderKernel.
Definition at line 40 of file NEFillBorder.h.
NEFillBorder | ( | ) |
Definition at line 34 of file NEFillBorder.cpp.
void configure | ( | ITensor * | input, |
unsigned int | border_width, | ||
BorderMode | border_mode, | ||
const PixelValue & | constant_border_value = PixelValue() |
||
) |
Initialize the function's source, destination and border_mode.
Valid data layouts:
Valid data type configurations:
src | dst |
---|---|
All | All |
[in,out] | input | Source tensor. Data type supported: All |
[in] | border_width | Width of the tensor border in pixels. |
[in] | border_mode | Strategy to use for borders. |
[in] | constant_border_value | (Optional) Constant value to use for borders if border_mode is set to CONSTANT. |
Definition at line 38 of file NEFillBorder.cpp.
References ARM_COMPUTE_LOG_PARAMS, and arm_compute::test::validation::input.
|
overridevirtual |
Run the kernels contained in the function.
For CPU kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 48 of file NEFillBorder.cpp.
References Window::DimZ, Scheduler::get(), and IScheduler::schedule().