|
ONE - On-device Neural Engine
|
Class to perform HashtableLookup operation. More...
#include <CLHashtableLookup.h>

Public Member Functions | |
| void | configure (const ICLTensor *lookups, const ICLTensor *keys, const ICLTensor *intput, ICLTensor *output, ICLTensor *hits) |
| Set the input and output tensors. | |
Class to perform HashtableLookup operation.
Definition at line 61 of file CLHashtableLookup.h.
| void CLHashtableLookup::configure | ( | const ICLTensor * | lookups, |
| const ICLTensor * | keys, | ||
| const ICLTensor * | intput, | ||
| ICLTensor * | output, | ||
| ICLTensor * | hits | ||
| ) |
Set the input and output tensors.
| [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.