Skip to content

Commit

Permalink
Add unittest for TRANSPOSE op in .tflite models
Browse files Browse the repository at this point in the history
  • Loading branch information
CNOCycle committed Apr 9, 2024
1 parent 922a29e commit 6e262ff
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions modules/dnn/test/test_tflite_importer.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -251,6 +251,11 @@ TEST_P(Test_TFLite, fully_connected) {
testLayer("fully_connected");
}

TEST_P(Test_TFLite, permute) {
testLayer("permutation_3d");
testLayer("permutation_4d");
}

INSTANTIATE_TEST_CASE_P(/**/, Test_TFLite, dnnBackendsAndTargets());

}} // namespace
Expand Down

0 comments on commit 6e262ff

Please sign in to comment.