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.cc286 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
288 float* batch_end = batch_start + input_sequence_size; in TEST()
290 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST()
315 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST_P() local
317 float* batch_end = batch_start + input_sequence_size; in TEST_P()
319 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P()
342 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST_P() local
344 float* batch_end = batch_start + input_sequence_size; in TEST_P()
346 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P()
Dbasic_rnn_test.cc265 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
268 for (int i = 0; i < input_sequence_size; i++) { in TEST()
294 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST_P() local
297 for (int i = 0; i < input_sequence_size; i++) { in TEST_P()
322 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST_P() local
325 for (int i = 0; i < input_sequence_size; i++) { in TEST_P()
Dbidirectional_sequence_rnn_test.cc877 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST_P() local
879 float* batch_end = batch_start + input_sequence_size; in TEST_P()
881 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P()
972 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST_P() local
974 float* batch_end = batch_start + input_sequence_size; in TEST_P()
976 rnn.SetInput(input_sequence_size, batch_start, batch_end); in TEST_P()
1127 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
1131 float* batch_start = endtoend_input + k * input_sequence_size; in TEST()
1132 float* batch_end = batch_start + input_sequence_size; in TEST()
1330 const int input_sequence_size = rnn.input_size() * rnn.sequence_len(); in TEST() local
[all …]
Dlstm_test.cc1749 const int input_sequence_size = lstm_input.size(); in TEST() local
1750 EXPECT_GT(input_sequence_size, 0); in TEST()
1751 for (int i = 0; i < input_sequence_size; ++i) { in TEST()
1922 const int input_sequence_size = lstm_input.size(); in TEST() local
1923 EXPECT_GT(input_sequence_size, 0); in TEST()
1924 for (int i = 0; i < input_sequence_size; ++i) { in TEST()
2089 const int input_sequence_size = lstm_input.size(); in TEST() local
2090 EXPECT_GT(input_sequence_size, 0); in TEST()
2091 for (int i = 0; i < input_sequence_size; ++i) { in TEST()
2396 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
[all …]
Dunidirectional_sequence_lstm_test.cc396 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
397 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
400 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
412 const float* batch_end = batch_start + input_sequence_size * num_inputs; in VerifyGoldens()
414 lstm->SetInput(b * input_sequence_size * num_inputs, batch_start, in VerifyGoldens()
426 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
438 golden_batch_start + input_sequence_size * num_outputs; in VerifyGoldens()
2519 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
2520 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
2522 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
[all …]
Dsvdf_test.cc242 const int input_sequence_size = in VerifyGoldens() local
246 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
Dbidirectional_sequence_lstm_test.cc2584 const int input_sequence_size = lstm.sequence_length() * lstm.num_inputs(); in TEST() local
2585 EXPECT_EQ(input_sequence_size, 20); in TEST()
2587 float* batch0_end = batch0_start + input_sequence_size; in TEST()
2591 float* batch1_end = batch1_start + input_sequence_size; in TEST()
2592 lstm.SetInput(input_sequence_size, batch1_start, batch1_end); in TEST()
Dfully_connected_test.cc1159 const int input_sequence_size = sizeof(fully_connected_input) / in TEST_P() local
1162 for (int i = 0; i < input_sequence_size; i++) { in TEST_P()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc2800 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
2803 for (int i = 0; i < input_sequence_size; i++) { in TEST()
3002 const int input_sequence_size = in VerifyGoldens() local
3006 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
3380 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
3381 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
3382 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()