24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NEELEMENTWISEOPERATIONS_H
25 #define ARM_COMPUTE_NEELEMENTWISEOPERATIONS_H
99 std::unique_ptr<Impl> _impl;
165 std::unique_ptr<Impl> _impl;
231 std::unique_ptr<Impl> _impl;
293 std::unique_ptr<Impl> _impl;
356 std::unique_ptr<Impl> _impl;
418 std::unique_ptr<Impl> _impl;
426 template <ComparisonOperation op>
464 std::unique_ptr<Impl> _impl;
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
~NEElementwiseComparison()
Default Destructor.
Basic function to run cpu::kernels::CpuComparisonKernel.
~NEElementwiseSquaredDiff()
Default Destructor.
Base class for all functions.
void run() override
Run the kernels contained in the function.
NEElementwisePower()
Default Constructor.
void run() override
Run the kernels contained in the function.
~NEElementwiseMin()
Default Destructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
Interface for CPU tensor.
void run() override
Run the kernels contained in the function.
NEElementwiseMin()
Default Constructor.
NEElementwiseComparison & operator=(const NEElementwiseComparison &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to run cpu::kernels::CpuArithmeticKernel for max.
NEElementwiseSquaredDiff()
Default Constructor.
~NEElementwiseMax()
Default Destructor.
Activation Layer Information class.
~NEElementwiseDivision()
Default Destructor.
void run() override
Run the kernels contained in the function.
NEElementwiseMin & operator=(const NEElementwiseMin &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Basic function to run cpu::kernels::CpuComparisonKernel.
ComparisonOperation
Supported comparison operations.
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 squared difference.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
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...
void run() override
Run the kernels contained in the function.
Basic function to run cpu::kernels::CpuArithmeticKernel for power.
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...
NEElementwiseMax & operator=(const NEElementwiseMax &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEElementwiseSquaredDiff & operator=(const NEElementwiseSquaredDiff &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEElementwisePower & operator=(const NEElementwisePower &)=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, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuArithme...
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
NEElementwiseComparisonStatic()
Default Constructor.
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...
void configure(ITensor *input1, ITensor *input2, ITensor *output, ComparisonOperation op)
Initialise the kernel's inputs, output and conversion policy.
Copyright (c) 2017-2024 Arm Limited.
Basic function to run cpu::kernels::CpuArithmeticKernel for division.
NEElementwiseComparison()
Default Constructor.
NEElementwiseDivision()
Default Constructor.
void configure(ITensor *input1, ITensor *input2, ITensor *output, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs, output and conversion policy.
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.
Store the tensor's metadata.
void run() override
Run the kernels contained in the function.
void run() override
Run the kernels contained in the function.
NEElementwiseComparisonStatic & operator=(const NEElementwiseComparisonStatic &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEElementwiseDivision & operator=(const NEElementwiseDivision &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
NEElementwiseMax()
Default Constructor.
~NEElementwisePower()
Default Destructor.
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...
void configure(ITensor *input1, ITensor *input2, ITensor *output)
Initialise the kernel's inputs, output and conversion policy.
Basic function to run cpu::kernels::CpuArithmeticKernel for min.