Browse Source

!224 add namespace parser for SWITCH&REFSWITCH

Merge pull request !224 from yangyongqiang/dev_parser
pull/224/MERGE
i-robot Gitee 5 years ago
parent
commit
b1e97fc5db
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      parser/tensorflow/tensorflow_parser.cc

+ 2
- 2
parser/tensorflow/tensorflow_parser.cc View File

@@ -3241,12 +3241,12 @@ Status TensorFlowModelParser::AddControlEdgeAfterRemoveInputs(domi::tensorflow::
return FAILED;
}
NodeDef *input_node_def = it->second;
if (input_node_def->op() == SWITCH || input_node_def->op() == REFSWITCH) {
if (input_node_def->op() == parser::SWITCH || input_node_def->op() == parser::REFSWITCH) {
NodeDef *identity_node_def = graph_def->add_node();
GE_CHECK_NOTNULL(identity_node_def);
input_node_name = input_node_name + "identity";
identity_node_def->set_name(input_node_name);
identity_node_def->set_op(IDENTITY);
identity_node_def->set_op(parser::IDENTITY);
identity_node_def->add_input(remove_input);
}
string control_input = "^" + input_node_name;


Loading…
Cancel
Save