24 #ifndef ARM_COMPUTE_CPU_CONCATENATE_H 25 #define ARM_COMPUTE_CPU_CONCATENATE_H 77 std::vector<std::unique_ptr<ICpuKernel>> _concat_kernels;
78 unsigned int _num_srcs;
void configure(const std::vector< const ITensorInfo *> &srcs_vector, ITensorInfo *dst, size_t axis)
Configure operator for a given list of arguments.
Basic interface for functions which have a single async Neon kernel.
Store the tensor's metadata.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const std::vector< const ITensorInfo *> &srcs_vector, const ITensorInfo *dst, size_t axis)
Static function to check if given info will lead to a valid configuration of NEConcatenateLayer.
CpuConcatenate()
Constructor.
Basic function to execute concatenate tensors along a given axis.