Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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()
H A 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 …]
H A 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()
H A 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 …]
H A Dunidirectional_sequence_lstm_test.cc144 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
145 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
148 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
160 const float* batch_end = batch_start + input_sequence_size * num_inputs; in VerifyGoldens()
162 lstm->SetInput(b * input_sequence_size * num_inputs, batch_start, in VerifyGoldens()
174 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
186 golden_batch_start + input_sequence_size * num_outputs; in VerifyGoldens()
2267 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
2268 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
2270 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()
[all …]
H A Dsvdf_test.cc242 const int input_sequence_size = in VerifyGoldens() local
246 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
H A 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()
H A Dfully_connected_test.cc1515 const int input_sequence_size = sizeof(fully_connected_input) / in TEST_P() local
1518 for (int i = 0; i < input_sequence_size; i++) { in TEST_P()
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp359 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST() local
360 for (int i = 0; i < input_sequence_size; i++) { in TEST()
456 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST() local
457 for (int i = 0; i < input_sequence_size; i++) { in TEST()
984 const int input_sequence_size = sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST() local
985 for (int i = 0; i < input_sequence_size; i++) { in TEST()
DSVDFTest.cpp341 const int input_sequence_size = in TEST() local
345 for (int i = 0; i < input_sequence_size; i++) { in TEST()
400 const int input_sequence_size = in TEST() local
404 for (int i = 0; i < input_sequence_size; i++) { in TEST()
DRNNTest.cpp283 const int input_sequence_size = in TEST() local
286 for (int i = 0; i < input_sequence_size; i++) { in TEST()
DLayerNormLSTMTest.cpp405 const int input_sequence_size = lstm_input[0].size() / n_input; in TEST() local
406 for (int i = 0; i < input_sequence_size; i++) { in TEST()
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_1/
Drnn_relaxed.mod.py185 input_sequence_size = int(len(test_inputs) / input_size / batches) variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_2/
Drnn_float16.mod.py184 input_sequence_size = int(len(test_inputs) / input_size / batches) variable
/aosp_15_r20/packages/modules/NeuralNetworks/runtime/test/specs/V1_0/
Drnn.mod.py184 input_sequence_size = int(len(test_inputs) / input_size / batches) variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate_test.cc2877 const int input_sequence_size = sizeof(rnn_input) / sizeof(float) / in TEST() local
2880 for (int i = 0; i < input_sequence_size; i++) { in TEST()
3079 const int input_sequence_size = in VerifyGoldens() local
3083 for (int i = 0; i < input_sequence_size; i++) { in VerifyGoldens()
3457 const int input_sequence_size = input[0].size() / num_inputs; in VerifyGoldens() local
3458 EXPECT_GT(input_sequence_size, 0); in VerifyGoldens()
3459 for (int i = 0; i < input_sequence_size; ++i) { in VerifyGoldens()