40 : _space_to_batch_kernel(), _fill_f(), _has_padding(false)
51 _fill_f = std::make_unique<NEFill>();
54 _space_to_batch_kernel = std::make_unique<NESpaceToBatchLayerKernel>();
55 _space_to_batch_kernel->configure(input, block_shape, paddings, output);
65 _fill_f = std::make_unique<NEFill>();
68 _space_to_batch_kernel = std::make_unique<NESpaceToBatchLayerKernel>();
69 _space_to_batch_kernel->configure(input, block_shape_x, block_shape_y, padding_left, padding_right, output);
Class describing the value of a pixel for any image format.
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
virtual DataType data_type() const =0
Data type used for each element of the 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.
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 NESpaceToBatchLayerKerne...
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.
virtual const TensorShape & tensor_shape() const =0
Size for each dimension of the tensor.
~NESpaceToBatchLayer()
Default destructor.
size_t total_size() const
Collapses all dimensions to a single linear total size.
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
virtual QuantizationInfo quantization_info() const =0
Get the quantization settings (scale and offset) of the tensor.
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
virtual void schedule(ICPPKernel *kernel, const Hints &hints)=0
Runs the kernel in the same thread as the caller synchronously.
Class for specifying the size of an image or rectangle.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
void run() override
Run the kernels contained in the function.
static IScheduler & get()
Access the scheduler singleton.