24 #ifndef ARM_COMPUTE_NESPACETOBATCHLAYER_H 25 #define ARM_COMPUTE_NESPACETOBATCHLAYER_H 36 class NESpaceToBatchLayerKernel;
104 std::unique_ptr<NESpaceToBatchLayerKernel> _space_to_batch_kernel;
105 std::unique_ptr<NEFill> _fill_f;
Base class for all functions.
Basic function to spatial divide a tensor.
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.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
NESpaceToBatchLayer()
Default constructor.
void configure(const ITensor *input, const ITensor *block_shape, const ITensor *paddings, ITensor *output)
Set the input and output tensors.
~NESpaceToBatchLayer()
Default destructor.
NESpaceToBatchLayer & operator=(const NESpaceToBatchLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Class for specifying the size of an image or rectangle.
void run() override
Run the kernels contained in the function.