Class to perform EmbeddingLookup operation. More...

Collaboration diagram for arm_compute::NEEmbeddingLookup:

Public Member Functions
void	configure (const ITensor input, ITensor output, const ITensor *lookups)
	Set the input and output tensors.

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

Detailed Description

Class to perform EmbeddingLookup operation.

Definition at line 63 of file NEEmbeddingLookup.h.

Member Function Documentation

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

Set the input and output tensors.

Parameters

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

Definition at line 47 of file NEEmbeddingLookup.cpp.

{
  auto k = std::make_unique<NEEmbeddingLookupKernel>();
  k->configure(input, output, lookups);
  _kernel = std::move(k);
}

static Status arm_compute::NEEmbeddingLookup::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 NECopy

Parameters

[in]	input	Source tensor info. Data types supported: U8/S8/QASYMM8/U16/S16/F16/U32/S32/F32.
[in]	output	Output tensor info. Data types supported: Same as `input`.
[in]	output	Lookups tensor info. Data types supported: S32.

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

runtime/compute/ARMComputeEx/arm_compute/runtime/NEON/functions/NEEmbeddingLookup.h
runtime/compute/ARMComputeEx/src/runtime/NEON/functions/NEEmbeddingLookup.cpp