Searched refs:rnn (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNNTest.cpp | 239 BasicRNNOpModel rnn(2, 16, 8); in TEST() local
240 rnn.SetWeights( in TEST()
261 rnn.SetBias({0.065691948, -0.69055247, 0.1107955, -0.97084129, -0.23957068, -0.23566568, in TEST()
265 rnn.SetRecurrentWeights( in TEST()
281 rnn.ResetHiddenState(); in TEST()
283 sizeof(rnn_input) / sizeof(float) / (rnn.input_size() * rnn.num_batches()); in TEST()
286 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
287 float* batch_end = batch_start + rnn.input_size(); in TEST()
288 rnn.SetInput(0, batch_start, batch_end); in TEST()
289 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
[all …]
|
| /frameworks/ml/nn/runtime/test/generated/spec_V1_0/ |
| D | rnn.example.cpp | 7 namespace generated_tests::rnn { namespace
114 namespace generated_tests::rnn { namespace
|