Searched refs:lstm_input (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTMTest.cpp | 355 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
366 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
368 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
461 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
472 sizeof(lstm_input) / sizeof(float) / (lstm.num_inputs()); in TEST()
474 float* batch0_start = lstm_input + i * lstm.num_inputs(); in TEST()
1040 static float lstm_input[][20] = { in TEST() local
1086 sizeof(lstm_input[0]) / sizeof(float) / (lstm.num_inputs()); in TEST()
1088 float* batch0_start = lstm_input[0] + i * lstm.num_inputs(); in TEST()
1093 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()
|