| 3a9b7c8eb10692fc746a954bfbc4b53b774d267b |
|
10-Oct-2017 |
Yang Ni <yangni@google.com> |
Made embedding lookup take int lookups
Bug: 67459279
NN API only supported float tensors when this op was first added.
Now int tensors are allowed in the API and we have updated the
documentation saying the lookup indices are integers, we need
update the implementaion and tests of this op to match the
documentation.
Also reverted the input order to match TF Lite.
Test: NeuralNetworksTest
Test: adb shell
/data/nativetest64/embedding_lookup_test/embedding_lookup_test
Change-Id: Ie0db884ca772e8fb388876c3cef609016e753bf1
/frameworks/ml/nn/common/operations/EmbeddingLookup.cpp
|
| 820215d28bed6c90f696cde0f282445d16da432e |
|
05-Oct-2017 |
Miao Wang <miaowang@google.com> |
Allow fine grain verbose logging control through system property.
- Existing DEBUG and INFO change to the new mechanism, existing ERROR and
WARNING unchanged.
- By default, all verbose logging are disabled.
- setprop debug.nn.vlog 1 enables all verbose logging.
- setprop debug.nn.vlog "tag1 tag2 ..." only enable the selected tags.
- Available tags:
all: same as 1.
model: enable logging with tag MODEL.
compilation: enable logging with tag COMPILATION.
execution: enable logging with tag EXECUTION.
cpuexe: enable logging with tag CPUEXE.
manager: enable logging with tag MANAGER.
driver: enable logging with tag DRIVER.
Bug: 63905942
Test: mm
Test: NeuralNetworksTests pass
Test: manually set property to allow individual module to logging
verbosely
Test: tested with debug.nn.partition default -> 0 -> 1 to make
sure getProp still works fine
Change-Id: Iaa0ffa62176dabcdc35058748597df720fd6e47e
/frameworks/ml/nn/common/operations/EmbeddingLookup.cpp
|
| 8c689bdfc391e47854ec27bad0f9d685135af253 |
|
27-Sep-2017 |
Yang Ni <yangni@google.com> |
Add preparation code missed in some OPs
Bug: 63905942
This updates speech model related OPs: Embedding Lookup,
Hashtable Lookup, LSH Projection, LSTM, RNN, and SVDF.
Deduce the shape of an output or intermediate tensor.
Allocate memory if needed (i.e., not set by user).
Validate inputs (dimensions, etc.)
Minor cleanups related to the utility functions shared by these ops.
Also made necessary fixes and implementations as discovered while
doing this.
Test: NeuralNetworksTest
Change-Id: I7c0b789edb4d3565526d58905609d2023c1a5351
/frameworks/ml/nn/common/operations/EmbeddingLookup.cpp
|
| 74a497bb146234759d84c7e4e4d9ae9c87c87986 |
|
12-Sep-2017 |
Yang Ni <yangni@google.com> |
Calculate storage size w/o assuming data type
Bug: 63905942
This allows the affected OPs to handle tensors of any data type, rather
than unnecessarily restricting them to a certain data type.
Updated OPs are Embedding Lookup, Hashtable Lookup, and LSH Projection.
Test: ran unit tests
Change-Id: I8b458a95249ad79204778b212a1f0b1146031913
/frameworks/ml/nn/common/operations/EmbeddingLookup.cpp
|
| 0831219b2763e7906d0f42bd15351a8e05c6f0ab |
|
01-Sep-2017 |
Yang Ni <yangni@google.com> |
Move operation files into subdirectory
Bug: 63905942
The convention seems to have all CPU implmentation of OPs in the
subdirectory nn/common/operations.
This CL moves recently added speech/text related OPs to that
directory.
Test: unit tests of all moved ops on marlin
Change-Id: I026aa94c58f6ee4e57e519781ee92d5792106de6
/frameworks/ml/nn/common/operations/EmbeddingLookup.cpp
|