24.02.1
|
Go to the documentation of this file.
25 #ifndef ACL_SRC_CORE_NEON_KERNELS_NESTACKLAYERKERNEL_H
26 #define ACL_SRC_CORE_NEON_KERNELS_NESTACKLAYERKERNEL_H
43 const char *
name()
const override
45 return "NEStackLayerKernel";
95 return _split_dimension;
99 std::vector<ITensor *> _input;
102 uint32_t _split_dimension;
104 std::function<void(
const std::vector<ITensor *> &,
ITensor *, uint32_t,
const Window &)> _stack_fn{};
107 #endif // ACL_SRC_CORE_NEON_KERNELS_NESTACKLAYERKERNEL_H
~NEStackLayerKernel()=default
Default destructor.
void prepare()
Prepare the reshape kernel for execution (Only executed once) for choosing the window and the algorit...
Basic kernel to stack a rank-R tensor into one with rank-(R+1) along the axis dimension.
Interface for CPU tensor.
static Status validate(const std::vector< ITensorInfo * > &input, uint32_t axis, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NEStackLayerKernel.
const char * name() const override
Name of the kernel.
NEStackLayerKernel()
Default constructor.
Common interface for all kernels implemented in C++.
uint32_t get_split_dimension() const
Get the dimension to split the kernel workload.
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
Describe a multidimensional execution window.
void configure(const std::vector< ITensor * > &input, uint32_t axis, ITensor *output)
Initialise the kernel's inputs and output.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
NEStackLayerKernel & operator=(const NEStackLayerKernel &)=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.