Merge pull request #64 from JDAI-CV/gemm

Support gemm in last layer
JDAI-CV · Oct 23, 2019 · 7b07333 · 7b07333
2 parents d1c6ad5 + dd4c7f8
commit 7b07333
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");
+2 −0		onnx/optimizer/pass_registry.h
+104 −0		onnx/optimizer/passes/dabnn_convert_gemm_with_reshape_or_flatten_to_conv_and_reshape.h