24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NECONCATENATELAYER_H
25 #define ARM_COMPUTE_NECONCATENATELAYER_H
76 void configure(std::vector<const ITensor *> inputs_vector,
ITensor *output,
size_t axis);
90 validate(
const std::vector<const ITensorInfo *> &inputs_vector,
const ITensorInfo *output,
size_t axis);
97 std::unique_ptr<Impl> _impl;
void configure(std::vector< const ITensor * > inputs_vector, ITensor *output, size_t axis)
Initialise the kernel's inputs vector and output.
Base class for all functions.
NEConcatenateLayer()
Default constructor.
Interface for CPU tensor.
NEConcatenateLayer & operator=(const NEConcatenateLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to execute concatenate tensors along a given axis.
static Status validate(const std::vector< const ITensorInfo * > &inputs_vector, const ITensorInfo *output, size_t axis)
Static function to check if given info will lead to a valid configuration of NEConcatenateLayer.
Copyright (c) 2017-2024 Arm Limited.
~NEConcatenateLayer()
Destructor.
Store the tensor's metadata.
void run() override
Run the kernels contained in the function.