Browse Source

Pre Merge pull request !620 from 王涛/ge_dev

pull/620/MERGE
王涛 Gitee 2 years ago
parent
commit
46b625775f
No known key found for this signature in database GPG Key ID: 173E9B9CA92EEF8F
2 changed files with 5 additions and 1 deletions
  1. +1
    -1
      metadef
  2. +4
    -0
      parser/tensorflow/tensorflow_constant_parser.cc

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit 3c1c28b3f0f31404933cfc1bd3cad28618b09317
Subproject commit 1c8737f98c45811dace5ae16367b947dadc55f1d

+ 4
- 0
parser/tensorflow/tensorflow_constant_parser.cc View File

@@ -28,6 +28,7 @@
#include "parser/common/op_parser_factory.h"
#include "framework/omg/parser/parser_types.h"
#include "register/tensor_assign.h"
#include "register/register_utils.h"

using domi::tensorflow::NodeDef;
using domi::TENSORFLOW;
@@ -90,6 +91,9 @@ Status TensorFlowConstantParser::ParseParams(const Message *op_src, ge::OpDescPt
const NodeDef *node = DOMI_DYNAMIC_CAST<const NodeDef *>(op_src);
GE_CHECK_NOTNULL(node);
GELOGD("TF op node name = %s, op type= %s, parse params", node->name().c_str(), node->op().c_str());
ge::Operator org_op = ge::OpDescUtils::CreateOperatorFromOpDesc(op_dest);
GE_CHK_STATUS_RET(domi::OperatorAutoMapping(op_src, org_op),
"[Call][AutoMapping] failed.");
ConstantOperator op;
op.Name(node->name());



Loading…
Cancel
Save