diff options
| author | Yang Ni <yangni@google.com> | 2017-09-29 14:07:17 -0700 |
|---|---|---|
| committer | Yang Ni <yangni@google.com> | 2017-09-29 15:52:18 -0700 |
| commit | 7da2c9a54b25fd15546d6f99eba06dff1ac9a862 (patch) | |
| tree | 2b36bfbb4e3be3e4fab9c94f386128964ef9903d /nn/runtime/include/NeuralNetworks.h | |
| parent | 47522fce0416bfae1cd26f676f251763cd22c172 (diff) | |
| download | ml-7da2c9a54b25fd15546d6f99eba06dff1ac9a862.tar.gz | |
Updated data type for keys in Hashtable Lookup
Bug: 63905942
This OP was initially added when the NN API only allowed floating-
point tensors.
Now since integer tensors are allowed in NN API, update the Hashtable
Lookup with the better suited data types.
Test: NeuralNetworksTest
Change-Id: I3dafb2f60bd428700a5a6de393b86dbd75f285dd
Diffstat (limited to 'nn/runtime/include/NeuralNetworks.h')
| -rw-r--r-- | nn/runtime/include/NeuralNetworks.h | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/nn/runtime/include/NeuralNetworks.h b/nn/runtime/include/NeuralNetworks.h index 1c456164c..9e5e7b666 100644 --- a/nn/runtime/include/NeuralNetworks.h +++ b/nn/runtime/include/NeuralNetworks.h @@ -421,8 +421,10 @@ typedef enum { * * Outputs: * * 0: Output. A tensor with shape [ k …]. - * * 1: Hits. A uint8 tensor with shape [ k ] indicates whether the lookup - * hits or not. + * * 1: Hits. A boolean tensor with shape [ k ] indicates whether the lookup + * hits (True) or not (False). + * Stored as {@link ANEURALNETWORKS_TENSOR_QUANT8_ASYMM} with offset 0 and scale 1.0f. + * A non-zero byte represents True, a hit. A zero indicates otherwise. */ ANEURALNETWORKS_HASHTABLE_LOOKUP = 10, |