| 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/HashtableLookup.cpp
|
| 7da2c9a54b25fd15546d6f99eba06dff1ac9a862 |
|
29-Sep-2017 |
Yang Ni <yangni@google.com> |
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
/frameworks/ml/nn/common/operations/HashtableLookup.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/HashtableLookup.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/HashtableLookup.cpp
|