提交 1df69f7c 编写于 作者: J Jacek Czaja

- Fix to comment

test=develop
上级 910cd415
...@@ -199,7 +199,7 @@ static int BuildFusion(Graph* graph, const std::string& name_scope, ...@@ -199,7 +199,7 @@ static int BuildFusion(Graph* graph, const std::string& name_scope,
embedding_lstm_creator(lookup_table, W, lstm, subgraph.at(x), w, Weight, embedding_lstm_creator(lookup_table, W, lstm, subgraph.at(x), w, Weight,
Bias, Hidden, Cell, fc_out, fc_bias); Bias, Hidden, Cell, fc_out, fc_bias);
// Remove unneeded nodes. // Remove unneeded nodes.
// TODO(jczaja): Proper removing of loopup table // TODO(jczaja): Proper removing of lookup table
std::unordered_set<const Node*> marked_nodes( std::unordered_set<const Node*> marked_nodes(
//{lookup_table, mul, lstm, elementwise_add, fc_bias, W}); //{lookup_table, mul, lstm, elementwise_add, fc_bias, W});
{mul, lstm, elementwise_add, fc_bias}); {mul, lstm, elementwise_add, fc_bias});
...@@ -209,7 +209,7 @@ static int BuildFusion(Graph* graph, const std::string& name_scope, ...@@ -209,7 +209,7 @@ static int BuildFusion(Graph* graph, const std::string& name_scope,
embedding_lstm_creator(lookup_table, W, lstm, subgraph.at(x), w, Weight, embedding_lstm_creator(lookup_table, W, lstm, subgraph.at(x), w, Weight,
Bias, Hidden, Cell, fc_out, nullptr); Bias, Hidden, Cell, fc_out, nullptr);
// Remove unneeded nodes. // Remove unneeded nodes.
// TODO(jczaja): Proper removing of loopup table // TODO(jczaja): Proper removing of lookup table
// std::unordered_set<const Node*> marked_nodes({lookup_table, W, mul, // std::unordered_set<const Node*> marked_nodes({lookup_table, W, mul,
// lstm}); // lstm});
std::unordered_set<const Node*> marked_nodes({mul, lstm}); std::unordered_set<const Node*> marked_nodes({mul, lstm});
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册