38 : _kernel(), _border_handler()
47 _kernel = std::make_unique<NEDerivativeKernel>();
48 _border_handler = std::make_unique<NEFillBorderKernel>();
51 _border_handler->configure(input,
BorderSize(1), border_mode,
PixelValue(constant_border_value));
BorderMode
Methods available to handle borders.
NEDerivative()
Default constructor.
Class describing the value of a pixel for any image format.
Container for 2D border size.
void configure(ITensor *input, ITensor *output_x, ITensor *output_y, BorderMode border_mode, uint8_t constant_border_value=0)
Initialise the function's source, destinations and border mode.
1 channel, 1 U8 per channel
#define ARM_COMPUTE_ERROR_ON(cond)
If the condition is true then an error message is printed and an exception thrown.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
#define ARM_COMPUTE_ERROR_ON_DATA_TYPE_CHANNEL_NOT_IN(t, c,...)
static constexpr size_t DimY
Alias for dimension 1 also known as Y dimension.
virtual void schedule(ICPPKernel *kernel, const Hints &hints)=0
Runs the kernel in the same thread as the caller synchronously.
static constexpr size_t DimZ
Alias for dimension 2 also known as Z dimension.
Borders are left undefined.
void run() override
Run the kernels contained in the function.
~NEDerivative()
Default destructor.
static IScheduler & get()
Access the scheduler singleton.