#include <NEEmbeddingLookupKernel.h>

Collaboration diagram for arm_compute::NEEmbeddingLookupKernel:

Public Member Functions
const char *	name () const override

	NEEmbeddingLookupKernel ()

	NEEmbeddingLookupKernel (const NEEmbeddingLookupKernel &)=delete

NEEmbeddingLookupKernel &	operator= (const NEEmbeddingLookupKernel &)=delete

	NEEmbeddingLookupKernel (NEEmbeddingLookupKernel &&)=default

NEEmbeddingLookupKernel &	operator= (NEEmbeddingLookupKernel &&)=default

void	configure (const ITensor input, ITensor output, const ITensor *lookups)

void	run (const Window &window, const ThreadInfo &info) override

Static Public Member Functions
static Status	validate (const ITensorInfo input, const ITensorInfo output, const ITensorInfo *lookups)

Detailed Description

NEON kernel to perform EmbeddingLookup operation

Definition at line 52 of file NEEmbeddingLookupKernel.h.

Constructor & Destructor Documentation

◆ NEEmbeddingLookupKernel() [1/3]

arm_compute::NEEmbeddingLookupKernel::NEEmbeddingLookupKernel ( )

Default constructor

◆ NEEmbeddingLookupKernel() [2/3]

arm_compute::NEEmbeddingLookupKernel::NEEmbeddingLookupKernel ( const NEEmbeddingLookupKernel & )

delete

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

◆ NEEmbeddingLookupKernel() [3/3]

arm_compute::NEEmbeddingLookupKernel::NEEmbeddingLookupKernel ( NEEmbeddingLookupKernel && )

default

Allow instances of this class to be moved

Member Function Documentation

◆ configure()

void arm_compute::NEEmbeddingLookupKernel::configure	(	const ITensor *	input,
		ITensor *	output,
		const ITensor *	lookups
	)

Initialize the kernel's input, output.

Parameters

[in]	input	Source tensor. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[out]	output	Destination tensor. Data types supported: same as `input`.
[in]	lookups	Lookups are 1D tensor that values are indices into the first dimension of input.

◆ name()

const char * arm_compute::NEEmbeddingLookupKernel::name ( ) const

inlineoverride

Definition at line 55 of file NEEmbeddingLookupKernel.h.

55{ return "NEEmbeddingLookupKernel"; }

◆ operator=() [1/2]

NEEmbeddingLookupKernel & arm_compute::NEEmbeddingLookupKernel::operator= ( const NEEmbeddingLookupKernel & )

delete

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

◆ operator=() [2/2]

NEEmbeddingLookupKernel & arm_compute::NEEmbeddingLookupKernel::operator= ( NEEmbeddingLookupKernel && )

default

Allow instances of this class to be moved

◆ run()

void arm_compute::NEEmbeddingLookupKernel::run	(	const Window &	window,
		const ThreadInfo &	info
	)

override

◆ validate()

static Status arm_compute::NEEmbeddingLookupKernel::validate	(	const ITensorInfo *	input,
		const ITensorInfo *	output,
		const ITensorInfo *	lookups
	)

static

Static function to check if given info will lead to a valid configuration of NEEmbeddingLookupKernel