24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CPU_ELEMENTWISE_UNARY_KERNEL_H
25 #define ARM_COMPUTE_CPU_ELEMENTWISE_UNARY_KERNEL_H
46 using ElementwiseUnaryUkernelPtr =
48 using ElementwiseUnaryPreparePtr = std::add_pointer<std::unique_ptr<uint8_t[]>(
72 const char *
name()
const override;
86 ElementwiseUnaryUkernelPtr _run_method{
nullptr};
88 std::unique_ptr<uint8_t[]> _lut{};
SimpleTensor< float > src
decltype(strategy::transforms) typedef type
void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override
Execute the kernel on the passed window.
Interface for CPU tensor.
void configure(ElementWiseUnary op, const ITensorInfo &src, ITensorInfo &dst)
Function to configure the CpuElementwiseUnaryKernel.
ElementwiseUnaryPreparePtr prepare_func
static Status validate(ElementWiseUnary op, const ITensorInfo &src, const ITensorInfo &dst)
Static function to check if given info will lead to a valid configuration.
std::add_pointer< bool(const DataTypeISASelectorData &data)>::type DataTypeISASelectorPtr
ARM_COMPUTE_DISALLOW_COPY_ALLOW_MOVE(CpuElementwiseUnaryKernel)
ElementwiseUnaryUkernelPtr ukernel
static const std::vector< ElementwiseUnaryKernel > & get_available_kernels()
const Window & window() const
The maximum window the kernel can be executed on.
Information about executing thread and CPU.
Describe a multidimensional execution window.
const DataTypeISASelectorPtr is_selected
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.
CpuElementwiseUnaryKernel()=default
Interface for an element-wise unary operation kernel.
const char * name() const override
Name of the kernel.
ScaleKernelInfo info(interpolation_policy, default_border_mode, PixelValue(), sampling_policy, false)
ElementWiseUnary
Available element wise unary operations.