34 auto k = std::make_unique<NEThresholdKernel>();
35 k->configure(input, output, info);
36 _kernel = std::move(k);
void configure(const ITensor *input, ITensor *output, const ThresholdKernelInfo &info)
Initialise the function's source, destination, thresholds and threshold type.
Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ThresholdKernelInfo &info)
Static function to check if given info will lead to a valid configuration of NEThresholdKernel.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const ThresholdKernelInfo &info)
Static function to check if given info will lead to a valid configuration of NEThreshold.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)