Class to perform HashtableLookup operation. More...

Collaboration diagram for arm_compute::CLHashtableLookup:

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

Detailed Description

Class to perform HashtableLookup operation.

Definition at line 61 of file CLHashtableLookup.h.

Member Function Documentation

void CLHashtableLookup::configure	(	const ICLTensor *	lookups,
		const ICLTensor *	keys,
		const ICLTensor *	intput,
		ICLTensor *	output,
		ICLTensor *	hits
	)

Set the input and output tensors.

Parameters

[in]	lookups	Lookups 1D tensor that values are indices into the first dimension of input.
[in]	keys	Keys 1D tensor. keys and input pair represent a map. Data types supported: S32
[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`.
[out]	hits	Hits 1D tensor. A boolean tensor that indicates whether the lookup hits (True) or not (False). Data types supported: U8/QASYMM8

Definition at line 47 of file CLHashtableLookup.cpp.

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

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