Browse Source

!582 identity op automapping

Merge pull request !582 from yanmingda/ge_dev
pull/585/head
yanmingda i-robot 3 years ago
parent
commit
82bc8d82d8
2 changed files with 1 additions and 1 deletions
  1. +1
    -0
      parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc
  2. +0
    -1
      parser/tensorflow/tensorflow_identity_parser.cc

+ 1
- 0
parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc View File

@@ -164,4 +164,5 @@ REGISTER_OP_PARSER_CREATOR(TENSORFLOW, REFEXIT, TensorFlowAutoMappingParserAdapt
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, CONSTANT, TensorFlowAutoMappingParserAdapter); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, CONSTANT, TensorFlowAutoMappingParserAdapter);
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, PARALLELCONCATSTART, TensorFlowAutoMappingParserAdapter); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, PARALLELCONCATSTART, TensorFlowAutoMappingParserAdapter);
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, BITCAST, TensorFlowAutoMappingParserAdapter); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, BITCAST, TensorFlowAutoMappingParserAdapter);
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, IDENTITY, TensorFlowAutoMappingParserAdapter);
} }

+ 0
- 1
parser/tensorflow/tensorflow_identity_parser.cc View File

@@ -25,6 +25,5 @@ using ge::parser::IDENTITY;
using ge::parser::READVARIABLEOP; using ge::parser::READVARIABLEOP;


namespace ge { namespace ge {
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, IDENTITY, TensorFlowIdentityParser);
REGISTER_OP_PARSER_CREATOR(TENSORFLOW, READVARIABLEOP, TensorFlowIdentityParser); REGISTER_OP_PARSER_CREATOR(TENSORFLOW, READVARIABLEOP, TensorFlowIdentityParser);
} // namespace ge } // namespace ge

Loading…
Cancel
Save