25 #ifndef __ARM_COMPUTE_NEDEPTHCONCATENATEKERNEL_H__ 26 #define __ARM_COMPUTE_NEDEPTHCONCATENATEKERNEL_H__ 40 const char *
name()
const override 42 return "NEDepthConcatenateLayerKernel";
73 using DepthConcatFunction = void(
const ITensor *in,
ITensor *out, std::pair<int, int> start_xy,
int depth_offset,
const Window &window);
76 DepthConcatFunction *_func;
81 unsigned int _depth_offset;
void run(const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
Container for 2D border size.
Common interface for all kernels implemented in C++.
NEDepthConcatenateLayerKernel()
Default constructor.
~NEDepthConcatenateLayerKernel()=default
Default destructor.
NEDepthConcatenateLayerKernel & operator=(const NEDepthConcatenateLayerKernel &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
const char * name() const override
Name of the kernel.
Interface for NEON tensor.
src info() -> set_format(Format::S16)
This file contains all available output stages for GEMMLowp on OpenCL.
BorderSize border_size() const override
The size of the border for that kernel.
Interface for the depth concatenate kernel.
const Window & window() const
The maximum window the kernel can be executed on.
Describe a multidimensional execution window.
void configure(const ITensor *input, unsigned int depth_offset, ITensor *output)
Initialise the kernel's inputs and output.