Skip to content

GitLab

Commit 4e3d06ab authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 15db7a1e
Show whitespace changes
Inline Side-by-side
Showing with 12 additions and 13 deletions
src/rewrite_rnn.cpp
View file @ 4e3d06ab
...@@ -536,7 +536,7 @@ std::vector<instruction_ref> rewrite_rnn::gru_cell(bool is_forward, ...@@ -536,7 +536,7 @@ std::vector<instruction_ref> rewrite_rnn::gru_cell(bool is_forward,
instruction_ref zt{}; instruction_ref zt{};
instruction_ref ht{}; instruction_ref ht{};
if (bias != prog.end()) if(bias != prog.end())
{ {
// equation f(xt*(Wz^T) + Ht-1 * (Rz^T) + Wbz + Rbz) // equation f(xt*(Wz^T) + Ht-1 * (Rz^T) + Wbz + Rbz)
auto xt_wz = prog.insert_instruction(ins, op::dot{}, xt, tran_wz, wbz); auto xt_wz = prog.insert_instruction(ins, op::dot{}, xt, tran_wz, wbz);
...@@ -599,7 +599,6 @@ std::vector<instruction_ref> rewrite_rnn::gru_cell(bool is_forward, ...@@ -599,7 +599,6 @@ std::vector<instruction_ref> rewrite_rnn::gru_cell(bool is_forward,
xht_h = prog.insert_instruction(ins, op::add{}, xt_wh, rt_rh); xht_h = prog.insert_instruction(ins, op::add{}, xt_wh, rt_rh);
} }
ht = prog.insert_instruction(ins, actv_func2, xht_h); ht = prog.insert_instruction(ins, actv_func2, xht_h);
} }
// equation Ht = (1 - zt) (.) ht + zt (.) Ht-1 // equation Ht = (1 - zt) (.) ht + zt (.) Ht-1
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment