24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_SRC_CORE_NEON_KERNELS_NEDEPTHTOSPACELAYERKERNEL_H
25 #define ACL_SRC_CORE_NEON_KERNELS_NEDEPTHTOSPACELAYERKERNEL_H
38 const char *
name()
const override
40 return "NEDepthToSpaceLayerKernel";
82 size_t _split_dimension;
85 #endif // ACL_SRC_CORE_NEON_KERNELS_NEDEPTHTOSPACELAYERKERNEL_H
NEDepthToSpaceLayerKernel & operator=(const NEDepthToSpaceLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
DataLayout
[DataLayout enum definition]
Interface for the depth to space kernel.
Interface for CPU tensor.
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...
void configure(const ITensor *input, ITensor *output, int32_t block_shape)
Initialise the kernel's inputs and output.
const char * name() const override
Name of the kernel.
Common interface for all kernels implemented in C++.
size_t get_split_dimension() const
Get the dimension the scheduler should use to split.
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
~NEDepthToSpaceLayerKernel()=default
Default destructor.
Describe a multidimensional execution window.
Copyright (c) 2017-2024 Arm Limited.
NEDepthToSpaceLayerKernel()
Default constructor.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)