24 #ifndef ARM_COMPUTE_NEPADLAYER_H 25 #define ARM_COMPUTE_NEPADLAYER_H 39 class NEPadLayerKernel;
109 void configure_reflect_symmetric_mode(
ITensor *input,
ITensor *output);
113 std::unique_ptr<NEPadLayerKernel> _pad_kernel;
116 uint32_t _num_dimensions;
117 std::vector<NEStridedSlice> _slice_functions;
118 std::vector<NEConcatenateLayer> _concat_functions;
119 std::vector<Tensor> _slice_results;
120 std::vector<Tensor> _concat_results;
Class describing the value of a pixel for any image format.
Base class for all functions.
std::vector< PaddingInfo > PaddingList
List of padding information.
Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
PaddingMode
Padding mode to use for PadLayer.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const PaddingList &padding, const PixelValue constant_value=PixelValue(), const PaddingMode mode=PaddingMode::CONSTANT)
Static function to check if given info will lead to a valid configuration of NEPadLayer.
~NEPadLayer()
Default destructor.
NEPadLayer()
Default Constructor.
void run() override
Run the kernels contained in the function.
void configure(ITensor *input, ITensor *output, const PaddingList &padding, const PixelValue constant_value=PixelValue(), const PaddingMode mode=PaddingMode::CONSTANT)
Initialize the function.
Basic function to pad a tensor.
NEPadLayer & operator=(const NEPadLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Basic function to run cpu::kernels::CpuCopyKernel.