optimize the parse_gemm operator.

74d5607e · Shucai Xiao · 84290a10 · 74d5607e
Commit 74d5607e authored Mar 15, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

src/onnx/onnx.cpp src/onnx/onnx.cpp +2 -1

No files found.
--- a/src/onnx/onnx.cpp
+++ b/src/onnx/onnx.cpp
@@ -501,7 +501,8 @@ struct onnx_parser
                out_lens.back() = l2->get_shape().lens().back();
                auto l3         = args[2];
                if(!std::equal(
-                       out_lens.begin(), out_lens.end(), args[2]->get_shape().lens().begin()))
+                       out_lens.begin(), out_lens.end(), args[2]->get_shape().lens().begin()) &&
+                        out_lens.size() > 2)
                {
                    l3 = prog.add_instruction(op::multibroadcast{out_lens}, args[2]);
                }