24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_CONCATENATE_H
25 #define ARM_COMPUTE_CPU_CONCATENATE_H
70 std::vector<std::unique_ptr<ICPPKernel>> _concat_kernels{};
71 unsigned int _num_srcs{0};
72 unsigned int _axis{0};
Basic function to execute concatenate tensors along a given axis.
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 CPU kernel.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
Copyright (c) 2017-2024 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.
Store the tensor's metadata.