24.02.1
|
Go to the documentation of this file.
24 #ifndef ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NESTACKLAYER_H
25 #define ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NESTACKLAYER_H
37 class NEStackLayerKernel;
94 std::unique_ptr<NEStackLayerKernel> _stack_kernel;
98 #endif // ACL_ARM_COMPUTE_RUNTIME_NEON_FUNCTIONS_NESTACKLAYER_H
Base class for all functions.
Interface for CPU tensor.
NEStackLayer()
Default constructor.
void configure(const std::vector< ITensor * > &input, int axis, ITensor *output)
Initialise the kernel's inputs vector and output.
NEStackLayer & operator=(const NEStackLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
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.
Basic function to stack tensors along an axis.
Copyright (c) 2017-2024 Arm Limited.
~NEStackLayer()
Default destructor.
Store the tensor's metadata.
void run() override
Run the kernels contained in the function.