Browse Source

!612 sync ge_dev to master 20220728

Merge pull request !612 from 王涛/ge_dev
pull/616/MERGE
王涛 Gitee 2 years ago
parent
commit
c47c747ca5
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      parser/tensorflow/tensorflow_parser_register.h

+ 5
- 0
parser/tensorflow/tensorflow_parser_register.h View File

@@ -29,6 +29,7 @@
#include "parser/common/op_parser_factory.h"
#include "parser/tensorflow/tensorflow_op_parser.h"
#include "proto/tensorflow/node_def.pb.h"
#include "register/register_utils.h"

namespace ge {
class PARSER_FUNC_VISIBILITY TensorflowFinalizeable {
@@ -106,6 +107,10 @@ class PARSER_FUNC_VISIBILITY TensorflowOpParserAdapter : public TensorFlowOpPars
GELOGE(domi::FAILED, "Param is null");
return domi::FAILED;
}
ge::Operator op = ge::OpDescUtils::CreateOperatorFromOpDesc(op_dest);
GE_CHK_STATUS_RET(domi::OperatorAutoMapping(op_src, op),
"[Call][AutoMapping] failed.");
op.BreakConnect();
GE_RETURN_IF_ERROR(parse_params_fn_(node, param.get()));
param.get()->Name(node->name());
std::shared_ptr<ParserOperator> op_param = std::static_pointer_cast<ParserOperator>(param);


Loading…
Cancel
Save