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/runtime/test/specs/rnn.mod.py
|
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/runtime/test/specs/rnn.mod.py
|
bf15cf2352cd159679210c0e575a26e2e37b770a |
|
08-Sep-2017 |
Yang Ni <yangni@google.com> |
Test spec for the Basic RNN operation Bug: 63905942 Added test spec for RNN. Additional changes to make the generated test work: * Updated auto test harness to use EXPECT_NEAR with an absolute error of 1.0e-5 for comparison of floating point numbers. * Made the generate example (inputs and golden outputs) local variable to avoid issues in the second test run. Test: run generate_test.sh and run the generated tests. Change-Id: I8844fd134c6718707d63af0dc4b8447ce5257954
/frameworks/ml/nn/runtime/test/specs/rnn.mod.py
|