From c926c36d95e7dc20b1de5f927cc58aab3e0dc35c Mon Sep 17 00:00:00 2001 From: huanghui Date: Wed, 15 Apr 2020 16:21:53 +0800 Subject: [PATCH] fix export mslite model --- .../converter/lite_model/operations/conv2d_packer.cc | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/mindspore/ccsrc/predict/converter/lite_model/operations/conv2d_packer.cc b/mindspore/ccsrc/predict/converter/lite_model/operations/conv2d_packer.cc index 0fefb89c59..176b235f5f 100644 --- a/mindspore/ccsrc/predict/converter/lite_model/operations/conv2d_packer.cc +++ b/mindspore/ccsrc/predict/converter/lite_model/operations/conv2d_packer.cc @@ -28,8 +28,8 @@ bool Conv2dPacker(const CNodePtr &c_node_ptr, OpDefT *ms_op) { std::vector kernel_size_value = AnfAlgo::GetNodeAttr>(c_node_ptr, "kernel_size"); std::string kernel_pad_mode_value = AnfAlgo::GetNodeAttr(c_node_ptr, "pad_mode"); int kernel_pad_value = AnfAlgo::GetNodeAttr(c_node_ptr, "pad"); - int kernel_stride_value = AnfAlgo::GetNodeAttr(c_node_ptr, "stride"); - int kernel_dilation_value = AnfAlgo::GetNodeAttr(c_node_ptr, "dilation"); + auto kernel_stride_value = AnfAlgo::GetNodeAttr>(c_node_ptr, "stride"); + auto kernel_dilation_value = AnfAlgo::GetNodeAttr>(c_node_ptr, "dilation"); std::string kernel_data_format_value = AnfAlgo::GetNodeAttr(c_node_ptr, "data_format"); std::unique_ptr attr(new Conv2DT()); MS_EXCEPTION_IF_NULL(attr); @@ -43,15 +43,15 @@ bool Conv2dPacker(const CNodePtr &c_node_ptr, OpDefT *ms_op) { attr->channelOut = kernel_channel_value; attr->kernelW = kernel_size_value[0]; attr->kernelH = kernel_size_value[1]; - attr->strideW = kernel_stride_value; - attr->strideH = kernel_stride_value; + attr->strideW = kernel_stride_value[0]; + attr->strideH = kernel_stride_value[1]; attr->padMode = GetAttrPadMode(kernel_pad_mode_value); attr->padUp = kernel_pad_value; attr->padDown = kernel_pad_value; attr->padLeft = kernel_pad_value; attr->padRight = kernel_pad_value; - attr->dilateW = kernel_dilation_value; - attr->dilateH = kernel_dilation_value; + attr->dilateW = kernel_dilation_value[0]; + attr->dilateH = kernel_dilation_value[1]; attr->hasBias = false; ms_op->name = c_node_ptr->fullname_with_scope(); ms_op->attr.type = OpT_Conv2D;