From 4c4d5734aa8fbaa97c667c20cd20039b03b5d33d Mon Sep 17 00:00:00 2001
From: guohongzilong <2713219276@qq.com>
Date: Wed, 14 Oct 2020 10:33:50 +0800
Subject: [PATCH] import bn fusion

---
 mindspore/lite/src/ops/conv2d.cc | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/mindspore/lite/src/ops/conv2d.cc b/mindspore/lite/src/ops/conv2d.cc
index ba635eba09..ae22dbb2da 100644
--- a/mindspore/lite/src/ops/conv2d.cc
+++ b/mindspore/lite/src/ops/conv2d.cc
@@ -182,6 +182,16 @@ void Conv2D::PopulaterConv2DMultiGroup(const Primitive &prim, schema::PrimitiveT
   if (input_node->isa<Parameter>()) {
     auto param_node = input_node->cast<ParameterPtr>();
     ConvertConvWeight<float>(param_node);
+    auto abstractBase = param_node->abstract();
+    MS_ASSERT(abstractBase != nullptr);
+    if (utils::isa<abstract::AbstractTensorPtr>(abstractBase)) {
+      auto abstractTensor = utils::cast<abstract::AbstractTensorPtr>(abstractBase);
+      MS_ASSERT(abstractTensor != nullptr);
+      if (utils::isa<abstract::ShapePtr>(abstractTensor->BuildShape())) {
+        auto dims = utils::cast<abstract::ShapePtr>(abstractTensor->BuildShape())->shape();
+        attr->channelIn = dims[kAnfPopulaterOne];
+      }
+    }
   }
 
   primitive->value.type = schema::PrimitiveType_DepthwiseConv2D;