24 #ifndef ARM_COMPUTE_CLSTACKLAYER_H 25 #define ARM_COMPUTE_CLSTACKLAYER_H 35 class CLCompileContext;
36 class CLStackLayerKernel;
98 std::vector<ICLTensor *> _input;
99 std::vector<std::unique_ptr<CLStackLayerKernel>> _stack_kernels;
100 unsigned int _num_inputs;
void configure(const std::vector< ICLTensor *> &input, int axis, ICLTensor *output)
Initialise the kernel's inputs vector and output.
Base class for all functions.
void run() override
Run the kernels contained in the function.
CLStackLayer()
Default constructor.
Store the tensor's metadata.
Copyright (c) 2017-2021 Arm Limited.
CLStackLayer & operator=(const CLStackLayer &)=delete
Prevent instances of this class from being copied.
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 CLStackLayerKernel.
~CLStackLayer()
Default destructor.
Interface for OpenCL tensor.
Basic function to stack tensors along an axis.