Basic function to run NEBatchToSpaceLayerKernel.
More...
#include <NEBatchToSpaceLayer.h>
Basic function to run NEBatchToSpaceLayerKernel.
Definition at line 38 of file NEBatchToSpaceLayer.h.
◆ NEBatchToSpaceLayer() [1/3]
◆ NEBatchToSpaceLayer() [2/3]
Prevent instances of this class from being copied (As this class contains pointers)
◆ NEBatchToSpaceLayer() [3/3]
Prevent instances of this class from being moved (As this class contains non movable objects)
◆ ~NEBatchToSpaceLayer()
◆ configure() [1/2]
Set the input and output tensors.
- Parameters
-
[in] | input | Tensor input. Supported tensor rank: 4. Data types supported: All. |
[in] | block_shape | 1-D tensor with shape [M]. Data types supported: S32 |
[out] | output | Tensor output. Data types supported: same as input |
Definition at line 35 of file NEBatchToSpaceLayer.cpp.
37 auto k = std::make_unique<NEBatchToSpaceLayerKernel>();
38 k->configure(
input, block_shape, output);
39 _kernel = std::move(k);
◆ 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] | input | Tensor input. Supported tensor rank: 4. Data types supported: All. |
[in] | block_shape_x | Block shape x value. |
[in] | block_shape_y | Block shape y value. |
[out] | output | Tensor output. Data types supported: same as input |
Definition at line 42 of file NEBatchToSpaceLayer.cpp.
44 auto k = std::make_unique<NEBatchToSpaceLayerKernel>();
45 k->configure(
input, block_shape_x, block_shape_y, output);
46 _kernel = std::move(k);
◆ operator=() [1/2]
Prevent instances of this class from being copied (As this class contains pointers)
◆ operator=() [2/2]
Prevent instances of this class from being moved (As this class contains non movable objects)
◆ validate() [1/2]
Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer.
- Parameters
-
[in] | input | Tensor input info. Supported tensor rank: 4. Data types supported: All. |
[in] | block_shape | block shape tensor info with shape [M]. Data types supported: S32 |
[out] | output | Tensor output info. Data types supported: same as input |
- Returns
- a status
Definition at line 49 of file NEBatchToSpaceLayer.cpp.
References NEBatchToSpaceLayerKernel::validate().
Referenced by arm_compute::test::validation::DATA_TEST_CASE().
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...
◆ validate() [2/2]
Static function to check if given info will lead to a valid configuration of CLBatchToSpaceLayer (Static block shape).
- Parameters
-
[in] | input | Tensor input info. Supported tensor rank: 4. Data types supported: All. |
[in] | block_shape_x | Block shape x value. |
[in] | block_shape_y | Block shape y value. |
[out] | output | Tensor output info. Data types supported: same as input |
- Returns
- a status
Definition at line 54 of file NEBatchToSpaceLayer.cpp.
References NEBatchToSpaceLayerKernel::validate().
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...
The documentation for this class was generated from the following files: