Lines Matching refs:rnn
240 BasicRNNOpModel rnn(2, 16, 8); in TEST() local
241 rnn.SetWeights( in TEST()
262 rnn.SetBias({0.065691948, -0.69055247, 0.1107955, -0.97084129, -0.23957068, -0.23566568, in TEST()
266 rnn.SetRecurrentWeights( in TEST()
282 rnn.ResetHiddenState(); in TEST()
284 sizeof(rnn_input) / sizeof(float) / (rnn.input_size() * rnn.num_batches()); in TEST()
287 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
288 float* batch_end = batch_start + rnn.input_size(); in TEST()
289 rnn.SetInput(0, batch_start, batch_end); in TEST()
290 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
292 rnn.Invoke(); in TEST()
294 float* golden_start = rnn_golden_output + i * rnn.num_units(); in TEST()
295 float* golden_end = golden_start + rnn.num_units(); in TEST()
300 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()