Compute Library
 19.08
NEBatchToSpaceLayer Class Reference

Basic function to run NEBatchToSpaceLayerKernel. More...

#include <NEBatchToSpaceLayer.h>

Collaboration diagram for NEBatchToSpaceLayer:
[legend]

Public Member Functions

void configure (const ITensor *input, const ITensor *block_shape, ITensor *output)
 Set the input and output tensors. More...
 
void configure (const ITensor *input, int32_t block_shape_x, int32_t block_shape_y, ITensor *output)
 Set the input and output tensors. More...
 
- Public Member Functions inherited from INESimpleFunctionNoBorder
 INESimpleFunctionNoBorder ()
 Constructor. More...
 
void run () override final
 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 *block_shape, const ITensorInfo *output)
 Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer. More...
 
static Status validate (const ITensorInfo *input, int32_t block_shape_x, int32_t block_shape_y, const ITensorInfo *output)
 Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer (Static block shape). More...
 

Detailed Description

Basic function to run NEBatchToSpaceLayerKernel.

Definition at line 38 of file NEBatchToSpaceLayer.h.

Member Function Documentation

◆ configure() [1/2]

void configure ( const ITensor input,
const ITensor block_shape,
ITensor output 
)

Set the input and output tensors.

Parameters
[in]inputTensor input. Supported tensor rank: 4. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[in]block_shape1-D tensor with shape [M]. Data types supported: S32
[out]outputTensor output. Data types supported: same as input

Definition at line 34 of file NEBatchToSpaceLayer.cpp.

35 {
36  auto k = arm_compute::support::cpp14::make_unique<NEBatchToSpaceLayerKernel>();
37  k->configure(input, block_shape, output);
38  _kernel = std::move(k);
39 }

◆ configure() [2/2]

void configure ( const ITensor input,
int32_t  block_shape_x,
int32_t  block_shape_y,
ITensor output 
)

Set the input and output tensors.

(Static block shape).

Parameters
[in]inputTensor input. Supported tensor rank: 4. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[in]block_shape_xBlock shape x value.
[in]block_shape_yBlock shape y value.
[out]outputTensor output. Data types supported: same as input

Definition at line 41 of file NEBatchToSpaceLayer.cpp.

42 {
43  auto k = arm_compute::support::cpp14::make_unique<NEBatchToSpaceLayerKernel>();
44  k->configure(input, block_shape_x, block_shape_y, output);
45  _kernel = std::move(k);
46 }

◆ validate() [1/2]

Status validate ( const ITensorInfo input,
const ITensorInfo block_shape,
const ITensorInfo output 
)
static

Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer.

Parameters
[in]inputTensor input info. Supported tensor rank: 4. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[in]block_shapeblock shape tensor info with shape [M]. Data types supported: S32
[out]outputTensor output info. Data types supported: same as input
Returns
a status

Definition at line 48 of file NEBatchToSpaceLayer.cpp.

49 {
50  return NEBatchToSpaceLayerKernel::validate(input, block_shape, output);
51 }
static Status validate(const ITensorInfo *input, const ITensorInfo *block_shape, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NEBatchToSpaceLayerKerne...

References NEBatchToSpaceLayerKernel::validate().

◆ validate() [2/2]

Status validate ( const ITensorInfo input,
int32_t  block_shape_x,
int32_t  block_shape_y,
const ITensorInfo output 
)
static

Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer (Static block shape).

Parameters
[in]inputTensor input info. Supported tensor rank: 4. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[in]block_shape_xBlock shape x value.
[in]block_shape_yBlock shape y value.
[out]outputTensor output info. Data types supported: same as input
Returns
a status

Definition at line 53 of file NEBatchToSpaceLayer.cpp.

54 {
55  return NEBatchToSpaceLayerKernel::validate(input, block_shape_x, block_shape_y, output);
56 }
static Status validate(const ITensorInfo *input, const ITensorInfo *block_shape, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NEBatchToSpaceLayerKerne...

References NEBatchToSpaceLayerKernel::validate().


The documentation for this class was generated from the following files: