added test for lstm without hidden

states initialization
2 years ago · 3b1ee0549b
parent e3e1f704a4
commit 3b1ee0549b
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");