potok/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2026-07-31 08:13:04 +04:00
Code Packages Releases Wiki Activity

Merge pull request #21154 from pccvlab:MatMul_with_two_inputs

Browse Source 
Add BatchMatMul layer support for tf_importer

* two inputs

* support batch_matmul

* refactor: remove useless code

* refactor: decrease nesting
This commit is contained in:
Gruhuang
2021-12-10 19:44:27 +08:00
committed by GitHub
parent c08954c18b
commit 17bc8565f6
2 changed files with 27 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dnn/test/test_tf_importer.cpp
+8
View File
@@ -660,6 +660,14 @@ TEST_P(Test_TensorFlow_layers, matmul)
double l1 = target == DNN_TARGET_MYRIAD ? 6.1e-3 : default_l1;
runTensorFlowNet("nhwc_reshape_matmul", false, l1);
runTensorFlowNet("matmul_layout");
runTensorFlowNet("two_inputs_matmul");
}
TEST_P(Test_TensorFlow_layers, batch_matmul)
{
if (backend == DNN_BACKEND_OPENCV && target == DNN_TARGET_OPENCL_FP16)
applyTestTag(CV_TEST_TAG_DNN_SKIP_OPENCL_FP16);
runTensorFlowNet("batch_matmul");
}
TEST_P(Test_TensorFlow_layers, reshape)