Merge pull request #23545 from Abdurrahheem:tests_lstm_init_no_hidden_states

Added test for LSTM without hidden state initialisation
2 years ago · 6dbc5e032f
parent af1c63c0a0 3b1ee0549b
commit 6dbc5e032f
1 changed files with 7 additions and 0 deletions
--- a/modules/dnn/test/test_onnx_importer.cpp
+++ b/modules/dnn/test/test_onnx_importer.cpp
@ -1387,6 +1387,13 @@ TEST_P(Test_ONNX_layers, LSTM_cell_batchsize_5_seqlen_5)
    testONNXModels("lstm_cell_batchsize_5_seqlen_5", npy, 0, 0, false, false);
 }

+TEST_P(Test_ONNX_layers, LSTM_init_h0_c0)
+{
+    if(backend == DNN_BACKEND_CUDA)
+        applyTestTag(CV_TEST_TAG_DNN_SKIP_CUDA);
+    testONNXModels("lstm_init_h0_c0", npy, 0, 0, false, false, 3);
+}
+
 TEST_P(Test_ONNX_layers, Pad2d_Unfused)
 {
    testONNXModels("ReflectionPad2d");