Class to perform EmbeddingLookup operation. More...

Collaboration diagram for arm_compute::CLEmbeddingLookup:

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

Detailed Description

Class to perform EmbeddingLookup operation.

Definition at line 61 of file CLEmbeddingLookup.h.

Member Function Documentation

void CLEmbeddingLookup::configure	(	const ICLTensor *	input,
		ICLTensor *	output,
		const ICLTensor *	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.

Definition at line 46 of file CLEmbeddingLookup.cpp.

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

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

runtime/compute/ARMComputeEx/arm_compute/runtime/CL/functions/CLEmbeddingLookup.h
runtime/compute/ARMComputeEx/src/runtime/CL/functions/CLEmbeddingLookup.cpp