From 99a027eebdaf1bdcf7fcf3b9a92d7fff23aab6ae Mon Sep 17 00:00:00 2001 From: yan Date: Tue, 28 Jun 2022 16:07:26 +0800 Subject: [PATCH] automapping identity --- 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