Searched refs:lstm_input (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | LSTMTest.cpp | 350 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
359 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
361 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
447 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
456 const int input_sequence_size = sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
458 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
943 static float lstm_input[][20] = { in TEST() local
984 const int input_sequence_size = sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()
986 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()
991 float* batch1_start = lstm_input[1] + i * lstm.num_inputs(); in TEST()
|
| D | LayerNormLSTMTest.cpp | 375 const std::vector<std::vector<float>> lstm_input = { in TEST() local
405 const int input_sequence_size = lstm_input[0].size() / n_input; in TEST()
408 const float* batch_start = lstm_input[b].data() + i * n_input; in TEST()
|