From 82bc8d82d8111fc1aa9bcc1e2dea07721ae05c48 Mon Sep 17 00:00:00 2001 From: yanmingda Date: Tue, 28 Jun 2022 17:09:57 +0000 Subject: [PATCH] !582 identity op automapping Merge pull request !582 from yanmingda/ge_dev --- parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc | 1 + parser/tensorflow/tensorflow_identity_parser.cc | 1 - 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc index fba835d..3fc99de 100644 --- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc +++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc @@ -164,4 +164,5 @@ REGISTER_OP_PARSER_CREATOR(TENSORFLOW, REFEXIT, TensorFlowAutoMappingParserAdapt REGISTER_OP_PARSER_CREATOR(TENSORFLOW, CONSTANT, TensorFlowAutoMappingParserAdapter); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, PARALLELCONCATSTART, TensorFlowAutoMappingParserAdapter); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, BITCAST, TensorFlowAutoMappingParserAdapter); +REGISTER_OP_PARSER_CREATOR(TENSORFLOW, IDENTITY, TensorFlowAutoMappingParserAdapter); } diff --git a/parser/tensorflow/tensorflow_identity_parser.cc b/parser/tensorflow/tensorflow_identity_parser.cc index 03dee85..265ecc3 100644 --- a/parser/tensorflow/tensorflow_identity_parser.cc +++ b/parser/tensorflow/tensorflow_identity_parser.cc @@ -25,6 +25,5 @@ using ge::parser::IDENTITY; using ge::parser::READVARIABLEOP; namespace ge { -REGISTER_OP_PARSER_CREATOR(TENSORFLOW, IDENTITY, TensorFlowIdentityParser); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, READVARIABLEOP, TensorFlowIdentityParser); } // namespace ge