24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLSPACETOBATCHLAYERKERNEL_H
25 #define ARM_COMPUTE_CLSPACETOBATCHLAYERKERNEL_H
82 const int block_shape_x,
83 const int block_shape_y,
84 const Size2D &padding_left,
85 const Size2D &padding_right,
99 const int block_shape_x,
100 const int block_shape_y,
101 const Size2D &padding_left,
102 const Size2D &padding_right,
129 const int block_shape_x,
130 const int block_shape_y,
131 const Size2D &padding_left,
132 const Size2D &padding_right,
Interface for the space to batch kernel.
~CLSpaceToBatchLayerKernel()=default
Default destructor.
CLSpaceToBatchLayerKernel & operator=(const CLSpaceToBatchLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Interface for OpenCL tensor.
Class for specifying the size of an image or rectangle.
void configure(const ICLTensor *input, const ICLTensor *block_shape, const ICLTensor *paddings, ICLTensor *output)
Initialise the kernel's inputs and output.
CLSpaceToBatchLayerKernel()
Default constructor.
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 CLSpaceToBatchLayerKerne...
Common interface for all the OpenCL kernels.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
void run(const Window &window, cl::CommandQueue &queue) override
Enqueue the OpenCL kernel to process the given window on the passed OpenCL command queue.