arm_compute::NEHashtableLookup Class Reference

Class to perform HashtableLookup operation. More...

#include <NEHashtableLookup.h>

Collaboration diagram for arm_compute::NEHashtableLookup:

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

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

Detailed Description

Class to perform HashtableLookup operation.

Definition at line 63 of file NEHashtableLookup.h.

Member Function Documentation

configure()

void NEHashtableLookup::configure	(	const ITensor *	lookups,
		const ITensor *	keys,
		const ITensor *	input,
		ITensor *	output,
		ITensor *	hits
	)

Set the input and output tensors.

Parameters

[in]	lookups	Lookups 1D tensor that values are indices into the first dimension of input. Data types supported: S32
[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

Returns

N/A

Definition at line 47 of file NEHashtableLookup.cpp.

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

validate()

Status NEHashtableLookup::validate ( const ITensorInfo *  lookups, const ITensorInfo *  keys, const ITensorInfo *  input, const ITensorInfo *  output, const ITensorInfo *  hits  )

static

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

Parameters

[in]	lookups	Lookups 1D tensor info. Data types supported: S32
[in]	keys	Keys 1D tensor info. keys and input pair represent a map. Data types supported: S32
[in]	input	Source tensor info. Data types supported: U8/S8/QASYMM8/U16/S16/U32/S32/F16/F32
[in]	output	Destination tensor info. Data types and data layouts supported: Same as input.
[in]	hits	Hits 1D tensor info. A boolean tensor that indicates whether the lookup hits (True) or not (False). Data types supported: U8/QASYMM8

Returns

a status

Definition at line 55 of file NEHashtableLookup.cpp.

{
  return NEHashtableLookupKernel::validate(lookups, keys, input, output, hits);
}

References arm_compute::NEHashtableLookupKernel::validate().

---

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

• compute/ARMComputeEx/arm_compute/runtime/NEON/functions/NEHashtableLookup.h
• compute/ARMComputeEx/src/runtime/NEON/functions/NEHashtableLookup.cpp