|
|
|
@ -78,7 +78,7 @@ lite::STATUS WeightFormatHardCodePass::HardCodeONNX(const AnfNodePtr &conv_node,
|
|
|
|
|
// deconv (C x K/group x kH x kW) group = 1
|
|
|
|
|
// dedepth (C x K/group x kH x kW) group = channelIn ==> (C, multiplier, H, W)
|
|
|
|
|
if (op_type == schema::PrimitiveType_Conv2D || op_type == schema::PrimitiveType_DepthwiseConv2D ||
|
|
|
|
|
op_type == schema::PrimitiveType_DeConv2D) {
|
|
|
|
|
op_type == schema::PrimitiveType_DeConv2D || op_type == schema::PrimitiveType_DeDepthwiseConv2D) {
|
|
|
|
|
param_value->set_format(schema::Format::Format_KCHW);
|
|
|
|
|
} else {
|
|
|
|
|
MS_LOG(ERROR) << "Unsupported opType: " << EnumNamePrimitiveType(op_type)
|
|
|
|
|