From 0d8fe3e62407dbed40ca24e3f2dcd0b51949d060 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=D0=A1=D0=B5=D1=80=D0=B3=D0=B5=D0=B9=20=D0=91=D0=B0=D1=80?=
 =?utf8?q?=D0=B0=D0=BD=D0=BD=D0=B8=D0=BA=D0=BE=D0=B2/AI=20Tools=20Lab=20/S?=
 =?utf8?q?RR/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?=
 <s.barannikov@samsung.com>
Date: Wed, 4 Sep 2019 19:19:39 +0300
Subject: [PATCH] [mir_caffe2] Reshape bias for Convolution (#7170)

Fixes compilation error of mobilenet.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
---
 compiler/mir-caffe2-importer/caffe2_op_creator.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/compiler/mir-caffe2-importer/caffe2_op_creator.cpp b/compiler/mir-caffe2-importer/caffe2_op_creator.cpp
index af3825e..c26e2bd 100644
--- a/compiler/mir-caffe2-importer/caffe2_op_creator.cpp
+++ b/compiler/mir-caffe2-importer/caffe2_op_creator.cpp
@@ -392,7 +392,9 @@ Caffe2OpCreator::convertConv(const std::vector<mir::Operation::Output *> &inputs
 
   if (op.input_size() > 2)
   {
-    result = createOp<ops::AddOp>(result, inputs[2])->getOutput(0);
+    auto bias = inputs[2];
+    bias = createOp<ops::ReshapeOp>(bias, Shape{1, bias->getShape().dim(0), 1, 1})->getOutput(0);
+    result = createOp<ops::AddOp>(result, bias)->getOutput(0);
   }
 
   return {result};
-- 
2.7.4