Basic function to run NEDepthToSpaceLayerKernel.
More...
#include <NEDepthToSpaceLayer.h>
Basic function to run NEDepthToSpaceLayerKernel.
Definition at line 39 of file NEDepthToSpaceLayer.h.
◆ NEDepthToSpaceLayer() [1/3]
◆ NEDepthToSpaceLayer() [2/3]
Prevent instances of this class from being copied (As this class contains pointers)
◆ NEDepthToSpaceLayer() [3/3]
Prevent instances of this class from being moved (As this class contains non movable objects)
◆ ~NEDepthToSpaceLayer()
◆ configure()
void configure |
( |
const ITensor * |
input, |
|
|
ITensor * |
output, |
|
|
int32_t |
block_shape |
|
) |
| |
Set the input and output tensors.
- Parameters
-
[in] | input | Tensor input. Supported tensor rank: 4. Data types supported: All |
[out] | output | Tensor output. Data types supported: same as input |
[in] | block_shape | Block shape value. |
Definition at line 35 of file NEDepthToSpaceLayer.cpp.
37 auto k = std::make_unique<NEDepthToSpaceLayerKernel>();
38 k->configure(
input, output, block_shape);
39 _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()
Static function to check if given info will lead to a valid configuration of NEDepthToSpaceLayer.
- Parameters
-
[in] | input | Tensor input info. Supported tensor rank: 4. Data types supported: All |
[in] | output | Tensor output info. Data types supported: same as input |
[in] | block_shape | Block shape x value. |
- Returns
- a status
Definition at line 42 of file NEDepthToSpaceLayer.cpp.
References NEDepthToSpaceLayerKernel::validate().
static Status validate(const ITensorInfo *input, const ITensorInfo *output, int32_t block_shape)
Static function to check if given info will lead to a valid configuration of NEDepthToSpaceLayerKerne...
The documentation for this class was generated from the following files: