24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NESPACETOBATCHLAYER_H
25 #define ARM_COMPUTE_NESPACETOBATCHLAYER_H
36 class NESpaceToBatchLayerKernel;
86 const int block_shape_x,
87 const int block_shape_y,
88 const Size2D &padding_left,
89 const Size2D &padding_right,
116 const int block_shape_x,
117 const int block_shape_y,
118 const Size2D &padding_left,
119 const Size2D &padding_right,
126 std::unique_ptr<NESpaceToBatchLayerKernel> _space_to_batch_kernel;
127 std::unique_ptr<NEFill> _fill_f;
void configure(const ITensor *input, const ITensor *block_shape, const ITensor *paddings, ITensor *output)
Set the input and output tensors.
Base class for all functions.
Class for specifying the size of an image or rectangle.
Interface for CPU tensor.
void run() override
Run the kernels contained in the function.
Basic function to spatial divide a tensor.
~NESpaceToBatchLayer()
Default destructor.
NESpaceToBatchLayer & operator=(const NESpaceToBatchLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NESpaceToBatchLayer()
Default constructor.
Copyright (c) 2017-2024 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *block_shape, const ITensorInfo *paddings, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NESpaceToBatchLayer.
Store the tensor's metadata.