Lines Matching refs:rnn
266 BasicRNNOpModel rnn(2, 16, 8); in TEST() local
267 rnn.SetWeights( in TEST()
291 rnn.SetBias({0.065691948, -0.69055247, 0.1107955, -0.97084129, -0.23957068, in TEST()
296 rnn.SetRecurrentWeights({0.1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, in TEST()
313 rnn.ResetHiddenState(); in TEST()
315 (rnn.input_size() * rnn.num_batches()); in TEST()
318 float* batch_start = rnn_input + i * rnn.input_size(); in TEST()
319 float* batch_end = batch_start + rnn.input_size(); in TEST()
320 rnn.SetInput(0, batch_start, batch_end); in TEST()
321 rnn.SetInput(rnn.input_size(), batch_start, batch_end); in TEST()
323 rnn.Invoke(); in TEST()
325 float* golden_start = rnn_golden_output + i * rnn.num_units(); in TEST()
326 float* golden_end = golden_start + rnn.num_units(); in TEST()
331 EXPECT_THAT(rnn.GetOutput(), ElementsAreArray(ArrayFloatNear(expected))); in TEST()