| 8db6e487763be346310fc5468bd8f33f255ccb9c |
|
06-Oct-2017 |
Yang Ni <yangni@google.com> |
Adding filter rank to SVDF
Bug: 67597181
Pulled in the latest implementation from TF Lite.
This allows a rank parameter that is greater than 1.
Split the state into input and output states.
Added a test spec for rank 2 SVDF.
Test: NeuralNetworksTest
Test: SVDFTest: adb shell /data/nativetest64/svdf_test/svdf_test
Change-Id: I207cc081019369bf5e643e16e91ff664d10975cc
/frameworks/ml/nn/common/operations/SVDF.cpp
|
| eefb1e60444afd08a4350e11f281ac7064ebba63 |
|
04-Oct-2017 |
Yang Ni <yangni@google.com> |
Change stateful Ops to stateless ones
Bug: 63905942
Updated Ops RNN, LSTM, and SVDF.
Split outputs used for states into inputs and outputs.
Test: NeuralNetworksTest
Change-Id: Ia3d11f640cba4cab1b94d0b9746c46d347c024a4
/frameworks/ml/nn/common/operations/SVDF.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/SVDF.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/SVDF.cpp
|