24 #ifndef ARM_COMPUTE_NEELEMENTWISEOPERATIONS_H 25 #define ARM_COMPUTE_NEELEMENTWISEOPERATIONS_H 92 std::unique_ptr<Impl> _impl;
152 std::unique_ptr<Impl> _impl;
212 std::unique_ptr<Impl> _impl;
268 std::unique_ptr<Impl> _impl;
325 std::unique_ptr<Impl> _impl;
386 std::unique_ptr<Impl> _impl;
394 template <ComparisonOperation op>
432 std::unique_ptr<Impl> _impl;
void configure(ITensor *input1, ITensor *input2, ITensor *output, ComparisonOperation op)
Initialise the kernel's inputs, output and conversion policy.
void configure(ITensor *input1, ITensor *input2, ITensor *output)
Initialise the kernel's inputs, output and conversion policy.
Base class for all functions.
NEElementwiseSquaredDiff & operator=(const NEElementwiseSquaredDiff &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, ComparisonOperation op)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuCompari...
NEElementwiseMin()
Default Constructor.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
NEElementwiseComparisonStatic()
Default Constructor.
NEElementwiseMin & operator=(const NEElementwiseMin &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to run cpu::kernels::CpuArithmeticKernel for min.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
Basic function to run cpu::kernels::CpuArithmeticKernel for power.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
Store the tensor's metadata.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output)
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuCompari...
Basic function to run cpu::kernels::CpuComparisonKernel.
Basic function to run cpu::kernels::CpuArithmeticKernel for max.
Activation Layer Information class.
Interface for CPU tensor.
Copyright (c) 2017-2021 Arm Limited.
NEElementwiseComparison()
Default Constructor.
void run() override
Run the kernels contained in the function.
static Status validate(const ITensorInfo *input1, const ITensorInfo *input2, const ITensorInfo *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
~NEElementwiseComparisonStatic()
Default Destructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
~NEElementwiseMax()
Default Destructor.
void run() override
Run the kernels contained in the function.
Basic function to run cpu::kernels::CpuArithmeticKernel for division.
~NEElementwiseDivision()
Default Destructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
void run() override
Run the kernels contained in the function.
~NEElementwisePower()
Default Destructor.
ComparisonOperation
Supported comparison operations.
Basic function to run cpu::kernels::CpuArithmeticKernel for squared difference.
NEElementwiseMax()
Default Constructor.
NEElementwisePower & operator=(const NEElementwisePower &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
~NEElementwiseSquaredDiff()
Default Destructor.
void run() override
Run the kernels contained in the function.
Basic function to run cpu::kernels::CpuComparisonKernel.
NEElementwiseComparison & operator=(const NEElementwiseComparison &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
void run() override
Run the kernels contained in the function.
NEElementwiseSquaredDiff()
Default Constructor.
~NEElementwiseMin()
Default Destructor.
~NEElementwiseComparison()
Default Destructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
NEElementwisePower()
Default Constructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
NEElementwiseMax & operator=(const NEElementwiseMax &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEElementwiseDivision()
Default Constructor.
void run() override
Run the kernels contained in the function.
NEElementwiseDivision & operator=(const NEElementwiseDivision &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
NEElementwiseComparisonStatic & operator=(const NEElementwiseComparisonStatic &)=delete
Prevent instances of this class from being copied (As this class contains pointers)