|
|
|
@@ -189,7 +189,7 @@ void Conv2D::PopulaterConv2DMultiGroup(const Primitive &prim, schema::PrimitiveT |
|
|
|
attr->channelMultiplier = channel_mutiplier; |
|
|
|
|
|
|
|
MS_ASSERT(inputs.size() == kAnfPopulaterInputNumTwo); |
|
|
|
auto input_node = inputs[kAnfPopulaterInputNumOne]; |
|
|
|
auto input_node = inputs.at(kAnfPopulaterInputNumOne); |
|
|
|
MS_ASSERT(input_node != nullptr); |
|
|
|
if (input_node->isa<Parameter>()) { |
|
|
|
auto param_node = input_node->cast<ParameterPtr>(); |
|
|
|
@@ -201,7 +201,7 @@ void Conv2D::PopulaterConv2DMultiGroup(const Primitive &prim, schema::PrimitiveT |
|
|
|
MS_ASSERT(abstractTensor != nullptr); |
|
|
|
if (utils::isa<abstract::ShapePtr>(abstractTensor->BuildShape())) { |
|
|
|
auto dims = utils::cast<abstract::ShapePtr>(abstractTensor->BuildShape())->shape(); |
|
|
|
attr->channelIn = dims[kAnfPopulaterInputNumOne]; |
|
|
|
attr->channelIn = dims.at(kAnfPopulaterInputNumOne); |
|
|
|
} |
|
|
|
} |
|
|
|
} else if (input_node->isa<CNode>()) { |
|
|
|
|