Browse Source

!350 bugfix for update origin shape

Merge pull request !350 from yangyongqiang/origin_shape
pull/352/head
i-robot Gitee 4 years ago
parent
commit
ac43384652
2 changed files with 2 additions and 1 deletions
  1. +1
    -1
      metadef
  2. +1
    -0
      parser/onnx/onnx_data_parser.cc

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 077f633cc41c87f01bb42ec46cc54b481f0ac909
Subproject commit 407a3602015aad12870d28ce515a47f277748caf

+ 1
- 0
parser/onnx/onnx_data_parser.cc View File

@@ -48,6 +48,7 @@ Status OnnxDataParser::ParseParams(const Message *op_src, ge::Operator &op_def)

ge::TensorDesc tensor_desc;
tensor_desc.SetShape(ge::Shape(user_input_dims_v_));
tensor_desc.SetOriginShape(ge::Shape(user_input_dims_v_));
int64_t type = 1;
(void)op_def.GetAttr(ge::DATA_ATTR_NAME_DATA_TYPE, type);
tensor_desc.SetDataType(static_cast<DataType>(type));


Loading…
Cancel
Save