Searched defs:lstm_input (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | bidirectional_sequence_lstm_test.cc | 558 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST_P() local 724 static float lstm_input[] = {2., 3., 2., 3., 3., 4., 3., 4., 1., 1., 1., 1.}; in TEST_P() local 1039 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local 1816 static float lstm_input[][20] = { in TEST() local 2519 static float lstm_input[][20] = { in TEST() local 2758 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST_P() local 2932 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST_P() local
|
| D | lstm_test.cc | 1726 const std::vector<std::vector<float>> lstm_input = { in TEST() local 1899 const std::vector<std::vector<float>> lstm_input = { in TEST() local 2066 const std::vector<std::vector<float>> lstm_input = { in TEST() local
|
| D | unidirectional_sequence_lstm_test.cc | 2910 const std::vector<float> lstm_input = { in TEST() local 3079 const std::vector<float> lstm_input = { in TEST() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/calibration/ |
| D | calibrator_test.cc | 304 const std::vector<float> lstm_input = {0.3, 0.2}; in TEST() local 366 const std::vector<float> lstm_input = {0.3, 0.2, 0.9, 0.8}; in TEST() local 428 const std::vector<float> lstm_input = {0.3, 0.2, 0.9, 0.8}; in TEST() local
|
| /external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| D | Lstm.cpp | 50 std::tuple<Tensor, Tensor, Tensor> lstm_input( in lstm_input() function
|