Searched refs:input_sequence_size (Results 1 – 9 of 9) sorted by relevance
/external/tensorflow/tensorflow/lite/kernels/ |
D | unidirectional_sequence_rnn_test.cc | 283 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()
|
D | basic_rnn_test.cc | 264 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()
|
D | bidirectional_sequence_rnn_test.cc | 798 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()
|
D | unidirectional_sequence_lstm_test.cc | 349 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()
|
D | svdf_test.cc | 240 const int input_sequence_size = in VerifyGoldens() local 244 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
|
D | lstm_test.cc | 375 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()
|
D | bidirectional_sequence_lstm_test.cc | 2566 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()
|
D | fully_connected_test.cc | 856 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/ |
D | nnapi_delegate_test.cc | 1966 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()
|