Home
last modified time | relevance | path

Searched refs:input_sequence_size (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dunidirectional_sequence_rnn_test.cc283 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
285 float* batch_end = batch_start + input_sequence_size; in TEST()
287 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
308 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
310 float* batch_end = batch_start + input_sequence_size; in TEST()
312 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
334 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
336 float* batch_end = batch_start + input_sequence_size; in TEST()
338 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
Dbasic_rnn_test.cc264 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
267 for (int i = 0; i < input_sequence_size; i++) { in TEST()
291 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
294 for (int i = 0; i < input_sequence_size; i++) { in TEST()
319 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
322 for (int i = 0; i < input_sequence_size; i++) { in TEST()
Dbidirectional_sequence_rnn_test.cc798 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
800 float* batch_end = batch_start + input_sequence_size; in TEST()
802 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
874 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
876 float* batch_end = batch_start + input_sequence_size; in TEST()
878 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
1025 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
1029 float* batch_start = endtoend_input + k * input_sequence_size; in TEST()
1030 float* batch_end = batch_start + input_sequence_size; in TEST()
Dunidirectional_sequence_lstm_test.cc349 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
350 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
353 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
365 const float* batch_end = batch_start + input_sequence_size * num_inputs; in VerifyGoldens()
367 lstm->SetInput(b * input_sequence_size * num_inputs, batch_start, in VerifyGoldens()
379 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
391 golden_batch_start + input_sequence_size * num_outputs; in VerifyGoldens()
Dsvdf_test.cc240 const int input_sequence_size = in VerifyGoldens() local
244 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
Dlstm_test.cc375 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
376 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
377 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
1773 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
1774 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
1775 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
Dbidirectional_sequence_lstm_test.cc2566 const int input_sequence_size = lstm.sequence_length() * lstm.num_inputs(); in TEST() local
2567 EXPECT_EQ(input_sequence_size, 20); in TEST()
2569 float* batch0_end = batch0_start + input_sequence_size; in TEST()
2573 float* batch1_end = batch1_start + input_sequence_size; in TEST()
2574 lstm.SetInput(input_sequence_size, batch1_start, batch1_end); in TEST()
Dfully_connected_test.cc856 const int input_sequence_size = sizeof(fully_connected_input) / in TEST_P() local
859 for (int i = 0; i < input_sequence_size; i++) { in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1966 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
1969 for (int i = 0; i < input_sequence_size; i++) { in TEST()
2169 const int input_sequence_size = in VerifyGoldens() local
2173 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
2493 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
2494 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
2495 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()