Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A 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
H A 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
H A Dunidirectional_sequence_lstm_test.cc2910 const std::vector<float> lstm_input = { in TEST() local
3079 const std::vector<float> lstm_input = { in TEST() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DLSTMTest.cpp350 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
447 static float lstm_input[] = {2., 3., 3., 4., 1., 1.}; in TEST() local
943 static float lstm_input[][20] = { in TEST() local
DLayerNormLSTMTest.cpp375 const std::vector<std::vector<float>> lstm_input = { in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A 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
/aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/
H A DLstm.cpp50 std::tuple<Tensor, Tensor, Tensor> lstm_input( in lstm_input() function