| @@ -29,6 +29,7 @@ | |||||
| #include "parser/common/op_parser_factory.h" | #include "parser/common/op_parser_factory.h" | ||||
| #include "parser/tensorflow/tensorflow_op_parser.h" | #include "parser/tensorflow/tensorflow_op_parser.h" | ||||
| #include "proto/tensorflow/node_def.pb.h" | #include "proto/tensorflow/node_def.pb.h" | ||||
| #include "register/register_utils.h" | |||||
| namespace ge { | namespace ge { | ||||
| class PARSER_FUNC_VISIBILITY TensorflowFinalizeable { | class PARSER_FUNC_VISIBILITY TensorflowFinalizeable { | ||||
| @@ -106,6 +107,10 @@ class PARSER_FUNC_VISIBILITY TensorflowOpParserAdapter : public TensorFlowOpPars | |||||
| GELOGE(domi::FAILED, "Param is null"); | GELOGE(domi::FAILED, "Param is null"); | ||||
| return domi::FAILED; | 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())); | GE_RETURN_IF_ERROR(parse_params_fn_(node, param.get())); | ||||
| param.get()->Name(node->name()); | param.get()->Name(node->name()); | ||||
| std::shared_ptr<ParserOperator> op_param = std::static_pointer_cast<ParserOperator>(param); | std::shared_ptr<ParserOperator> op_param = std::static_pointer_cast<ParserOperator>(param); | ||||