24.02
|
Go to the documentation of this file.
51 const unsigned int axis_u =
wrap_around(axis,
static_cast<int>(
input[0]->
info()->num_dimensions() + 1));
53 _stack_kernel->configure(
input, axis_u, output);
62 const size_t rank =
input[0]->num_dimensions();
63 const unsigned int axis_u =
wrap_around(axis,
static_cast<int>(rank + 1));
75 _stack_kernel->prepare();
Basic kernel to stack a rank-R tensor into one with rank-(R+1) along the axis dimension.
Interface for CPU tensor.
NEStackLayer()
Default constructor.
T wrap_around(T x, T m)
Wrap-around a number within the range 0 <= x < m.
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.
void configure(const std::vector< ITensor * > &input, int axis, ITensor *output)
Initialise the kernel's inputs vector and output.
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
static Status validate(const std::vector< ITensorInfo * > &input, int axis, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of NEStackLayerKernel.
#define ARM_COMPUTE_RETURN_ERROR_ON(cond)
If the condition is true, an error is returned.
static IScheduler & get()
Access the scheduler singleton.
virtual void schedule(ICPPKernel *kernel, const Hints &hints)=0
Runs the kernel in the same thread as the caller synchronously.
Copyright (c) 2017-2024 Arm Limited.
~NEStackLayer()
Default destructor.
#define ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(...)
Store the tensor's metadata.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
#define ARM_COMPUTE_LOG_PARAMS(...)
void run() override
Run the kernels contained in the function.