From 419ba8b25ac5837ea8b4591e77a1c14718194213 Mon Sep 17 00:00:00 2001 From: y00500818 Date: Mon, 25 Jan 2021 09:02:11 +0800 Subject: [PATCH] add namespace parser for SWITCH&REFSWITCH --- parser/tensorflow/tensorflow_parser.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/parser/tensorflow/tensorflow_parser.cc b/parser/tensorflow/tensorflow_parser.cc index 00ac3e4..16557a6 100644 --- a/parser/tensorflow/tensorflow_parser.cc +++ b/parser/tensorflow/tensorflow_parser.cc @@ -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;