Support gemm in last layer

JDAI-CV · Oct 23, 2019 · dd4c7f8 · dd4c7f8
1 parent d1c6ad5
commit dd4c7f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/third_party/onnx b/third_party/onnx
diff --git a/tools/onnx2bnn/OnnxConverter.cpp b/tools/onnx2bnn/OnnxConverter.cpp
@@ -193,6 +193,7 @@ std::vector<std::string> OnnxConverter::Convert(
     // for details.
     vector<string> optimizers{"eliminate_nop_pad",
                               "extract_constant_to_initializer",
+                              "dabnn_convert_gemm_with_reshape_or_flatten_to_conv_and_reshape",
                               "dabnn_bconv_strict"};
     if (level == Level::kModerate || level == Level::kAggressive) {
         optimizers.push_back("dabnn_bconv_moderate");