Add comments for dabnn optimizer

JDAI-CV · May 15, 2019 · d55df70 · d55df70
1 parent da4e14a
commit d55df70
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tools/onnx2bnn/OnnxConverter.cpp b/tools/onnx2bnn/OnnxConverter.cpp
@@ -216,6 +216,12 @@ void OnnxConverter::Convert(const ONNX_NAMESPACE::ModelProto &model_proto,
                             const bool strict) {
     GOOGLE_PROTOBUF_VERIFY_VERSION;
 
+    // We recognize binary convolutions in our custom ONNX optimizers.
+    // Please check out 
+    // https://github.com/daquexian/onnx/blob/optimizer_for_bnn/onnx/optimizer/passes/dabnn_bconv_soft.h
+    // and
+    // https://github.com/daquexian/onnx/blob/optimizer_for_bnn/onnx/optimizer/passes/dabnn_bconv_strict.h
+    // for details.
     vector<string> optimizers{"eliminate_nop_pad", "dabnn_bconv_strict"};
     if (!strict) {
         optimizers.push_back("dabnn_bconv_soft");