21.02
|
Basic function to run GCNormalizePlanarYUVLayerKernel. More...
#include <GCNormalizePlanarYUVLayer.h>
Public Member Functions | |
GCNormalizePlanarYUVLayer () | |
Default constructor. More... | |
void | configure (const IGCTensor *input, IGCTensor *output, const IGCTensor *mean, const IGCTensor *std) |
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, const ITensorInfo *output, const ITensorInfo *mean, const ITensorInfo *std) |
Static function to check if given info will lead to a valid configuration of CLNormalizePlanarYUVLayer. More... | |
Basic function to run GCNormalizePlanarYUVLayerKernel.
Definition at line 43 of file GCNormalizePlanarYUVLayer.h.
Default constructor.
Definition at line 35 of file GCNormalizePlanarYUVLayer.cpp.
void configure | ( | const IGCTensor * | input, |
IGCTensor * | output, | ||
const IGCTensor * | mean, | ||
const IGCTensor * | std | ||
) |
Set the input and output tensors.
[in] | input | Source tensor. 3 lower dimensions represent a single input with dimensions [width, height, channels]. Data types supported: F16. |
[out] | output | Destination tensor. Data type supported: same as input |
[in] | mean | Mean values tensor. 1 dimension with size equal to the number of input channels. Data types supported: same as input |
[in] | std | Standard deviation values tensor. 1 dimension with size equal to the number of input channels. Data types supported: same as input |
Definition at line 40 of file GCNormalizePlanarYUVLayer.cpp.
References GCNormalizePlanarYUVLayerKernel::configure().
|
overridevirtual |
Run the kernels contained in the function.
For Neon kernels:
For OpenCL kernels:
Implements IFunction.
Definition at line 51 of file GCNormalizePlanarYUVLayer.cpp.
References GCScheduler::dispatch(), and GCScheduler::get().
|
static |
Static function to check if given info will lead to a valid configuration of CLNormalizePlanarYUVLayer.
[in] | input | Source tensor info. 3 lower dimensions represent a single input with dimensions [width, height, channels]. Data types supported: F16/F32. |
[out] | output | Destination tensor info. Data type supported: same as input |
[in] | mean | Mean values tensor info. 1 dimension with size equal to the number of input channels. Data types supported: same as input |
[in] | std | Standard deviation values tensor info. 1 dimension with size equal to the number of input channels. Data types supported: same as input |
Definition at line 45 of file GCNormalizePlanarYUVLayer.cpp.
References GCNormalizePlanarYUVLayerKernel::validate().
Referenced by arm_compute::test::validation::DATA_TEST_CASE().