Fix eltwise shaper

JDAI-CV · Jun 18, 2019 · 7c1d1b9 · 7c1d1b9
1 parent 38dd3bd
commit 7c1d1b9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/common/Shaper.cpp b/common/Shaper.cpp
@@ -243,8 +243,9 @@ void Shaper::Eltwise(const std::string &input1_name,
                      const std::string &output_name) {
     auto shape1 = shape_map_.at(input1_name);
     auto shape2 = shape_map_.at(input2_name);
+    // TODO: broadcasting
     auto output_shape =
-        shape1.size() > shape2.size() ? shape1 : shape2;  // broadcasting
+        shape1.size() >= shape2.size() ? shape1 : shape2;
     shape_map_[output_name] = output_shape;
 }