Browse Source

fix domi format not match with ge format

pull/311/head
lichun 4 years ago
parent
commit
cd1d9938bd
1 changed files with 6 additions and 4 deletions
  1. +6
    -4
      parser/onnx/onnx_parser.cc

+ 6
- 4
parser/onnx/onnx_parser.cc View File

@@ -41,6 +41,7 @@
#include "register/register_fmk_types.h"
#include "graph/utils/graph_utils.h"
#include "graph/utils/node_utils.h"
#include "graph/utils/type_utils.h"
#include "subgraph_adapter/subgraph_adapter_factory.h"

namespace ge {
@@ -1004,14 +1005,15 @@ void OnnxModelParser::UpdateDataFormat(ge::Graph &graph) {
}
TensorDesc in_desc;
gn.GetInputDesc(0, in_desc);
in_desc.SetOriginFormat(static_cast<ge::Format>(GetParserContext().format));
in_desc.SetFormat(static_cast<ge::Format>(GetParserContext().format));
ge::Format ge_format = TypeUtils::DomiFormatToFormat(GetParserContext().format);
in_desc.SetOriginFormat(ge_format);
in_desc.SetFormat(ge_format);
gn.UpdateInputDesc(0, in_desc);

TensorDesc out_desc;
gn.GetOutputDesc(0, out_desc);
out_desc.SetOriginFormat(static_cast<ge::Format>(GetParserContext().format));
out_desc.SetFormat(static_cast<ge::Format>(GetParserContext().format));
out_desc.SetOriginFormat(ge_format);
out_desc.SetFormat(ge_format);
gn.UpdateOutputDesc(0, out_desc);
}
GELOGD("Update data format success.");


Loading…
Cancel
Save