Home
last modified time | relevance | path

Searched defs:lstm_input (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dbidirectional_sequence_lstm_test.cc558 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
Dlstm_test.cc1726 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
Dunidirectional_sequence_lstm_test.cc2910 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/
Dcalibrator_test.cc304 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/
DLstm.cpp50 std::tuple<Tensor, Tensor, Tensor> lstm_input( in lstm_input() function