From 70920abf2f6970643e2dc93d17a02f50a2548690 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=A0=E7=9B=9B=E6=9D=B0?= Date: Thu, 28 Jul 2022 11:46:03 +0000 Subject: [PATCH] =?UTF-8?q?!607=20add=20automapping=20for=20variable=20etc?= =?UTF-8?q?.=20Merge=20pull=20request=20!607=20from=20=E5=BC=A0=E7=9B=9B?= =?UTF-8?q?=E6=9D=B0/add=5Fautomapping?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- parser/tensorflow/tensorflow_parser_register.h | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/parser/tensorflow/tensorflow_parser_register.h b/parser/tensorflow/tensorflow_parser_register.h index 17eaa23..fae07dc 100644 --- a/parser/tensorflow/tensorflow_parser_register.h +++ b/parser/tensorflow/tensorflow_parser_register.h @@ -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 op_param = std::static_pointer_cast(param);