m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

- Fix to comment

Browse Source 
test=develop
revert-13637-optimize-opyreader
Jacek Czaja 7 years ago
parent 910cd415f2
commit 1df69f7c9d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
paddle/fluid/framework/ir/embedding_fc_lstm_fuse_pass.cc
Unescape View File

@ -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,
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});

Write Preview
Loading…
Cancel
Save