NEElementwiseComparison Class Reference

Basic function to run cpu::kernels::CpuComparisonKernel. More...

#include <NEElementwiseOperations.h>

Collaboration diagram for NEElementwiseComparison:

Public Member Functions
	NEElementwiseComparison ()
	Default Constructor. More...
	~NEElementwiseComparison ()
	Default Destructor. More...
	NEElementwiseComparison (const NEElementwiseComparison &)=delete
	Prevent instances of this class from being copied (As this class contains pointers) More...
	NEElementwiseComparison (NEElementwiseComparison &&)
	Default move constructor. More...
NEElementwiseComparison &	operator= (const NEElementwiseComparison &)=delete
	Prevent instances of this class from being copied (As this class contains pointers) More...
NEElementwiseComparison &	operator= (NEElementwiseComparison &&)
	Default move assignment operator. More...
void	configure (ITensor *input1, ITensor *input2, ITensor *output, ComparisonOperation op)
	Initialise the kernel's inputs, output and conversion policy. More...
void	run () override
	Run the kernels contained in the function. More...
Public Member Functions inherited from IFunction
virtual	~IFunction ()=default
	Destructor. More...
virtual void	prepare ()
	Prepare the function for executing. More...

Static Public Member Functions
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::CpuComparisonKernel. More...

Detailed Description

Basic function to run cpu::kernels::CpuComparisonKernel.

Note

The tensor data type for the inputs must be U8/QASYMM8/QASYMM8_SIGNED/S16/F16/S32/F32.

The function performs a comparison operation between two tensors.

Definition at line 334 of file NEElementwiseOperations.h.

Constructor & Destructor Documentation

NEElementwiseComparison() [1/3]

NEElementwiseComparison

(

)

Default Constructor.

Definition at line 294 of file NEElementwiseOperations.cpp.

    : _impl(std::make_unique<Impl>())
{
}

~NEElementwiseComparison()

~NEElementwiseComparison ( )

default

Default Destructor.

NEElementwiseComparison() [2/3]

NEElementwiseComparison ( const NEElementwiseComparison &  )

delete

Prevent instances of this class from being copied (As this class contains pointers)

NEElementwiseComparison() [3/3]

NEElementwiseComparison ( NEElementwiseComparison &&  )

default

Default move constructor.

Member Function Documentation

configure()

void configure	(	ITensor *	input1,
		ITensor *	input2,
		ITensor *	output,
		ComparisonOperation	op
	)

Initialise the kernel's inputs, output and conversion policy.

Valid data layouts:

• All

Valid data type configurations:

src0	src1	dst
QASYMM8	QASYMM8	U8
QASYMM8_SIGNED	QASYMM8_SIGNED	U8
S32	S32	U8
U8	U8	U8
S16	S16	U8
F16	F16	U8
F32	F32	U8

Parameters

[in,out]	input1	First tensor input. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/F16/S32/F32.
[in,out]	input2	Second tensor input. Data types supported: Same as input1.
[out]	output	Output tensor. Data types supported: U8.
[in]	op	Comparison Operation to be performed.

Definition at line 302 of file NEElementwiseOperations.cpp.

{
    _impl->src_0 = input1;
    _impl->src_1 = input2;
    _impl->dst   = output;
    _impl->op    = std::make_unique<cpu::CpuElementwiseComparison>();
    _impl->op->configure(input1->info(), input2->info(), output->info(), op);
}

References ITensor::info().

operator=() [1/2]

NEElementwiseComparison& operator= ( const NEElementwiseComparison &  )

delete

Prevent instances of this class from being copied (As this class contains pointers)

operator=() [2/2]

NEElementwiseComparison & operator= ( NEElementwiseComparison &&  )

default

Default move assignment operator.

run()

void run ( )

overridevirtual

Run the kernels contained in the function.

For CPU kernels:

• Multi-threading is used for the kernels which are parallelisable.
• By default std::thread::hardware_concurrency() threads are used.

Note

CPPScheduler::set_num_threads() can be used to manually set the number of threads

For OpenCL kernels:

• All the kernels are enqueued on the queue associated with CLScheduler.
• The queue is then flushed.

Note

The function will not block until the kernels are executed. It is the user's responsibility to wait.

Will call prepare() on first run if hasn't been done

Implements IFunction.

Definition at line 316 of file NEElementwiseOperations.cpp.

{
    ITensorPack pack;
    pack.add_tensor(TensorType::ACL_SRC_0, _impl->src_0);
    pack.add_tensor(TensorType::ACL_SRC_1, _impl->src_1);
    pack.add_tensor(TensorType::ACL_DST, _impl->dst);
    _impl->op->run(pack);
}

References arm_compute::ACL_DST, arm_compute::ACL_SRC_0, arm_compute::ACL_SRC_1, ITensorPack::add_tensor(), and arm_compute::test::validation::pack.

validate()

Status validate ( const ITensorInfo *  input1, const ITensorInfo *  input2, const ITensorInfo *  output, ComparisonOperation  op  )

static

Static function to check if given info will lead to a valid configuration of cpu::kernels::CpuComparisonKernel.

Parameters

[in]	input1	First tensor input info. Data types supported: U8/QASYMM8/QASYMM8_SIGNED/S16/F16/S32/F32.
[in]	input2	Second tensor input info. Data types supported: Same as input1.
[in]	output	Output tensor info. Data types supported: U8.
[in]	op	Comparison Operation to be performed.

Returns

a status

Definition at line 311 of file NEElementwiseOperations.cpp.

{
    return cpu::CpuElementwiseComparison::validate(input1, input2, output, op);
}

References CpuElementwiseComparison::validate().

---

The documentation for this class was generated from the following files:

• arm_compute/runtime/NEON/functions/NEElementwiseOperations.h
• src/runtime/NEON/functions/NEElementwiseOperations.cpp