Commit 179fa5b2 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format.

parent 61566cb1
...@@ -207,7 +207,7 @@ std::vector<instruction_ref> rewrite_rnn::rnn_cell(bool is_forward, ...@@ -207,7 +207,7 @@ std::vector<instruction_ref> rewrite_rnn::rnn_cell(bool is_forward,
} }
instruction_ref hidden_out = prog.end(), last_out; instruction_ref hidden_out = prog.end(), last_out;
last_out = prog.insert_instruction(ins, op::unsqueeze{{0, 1}}, sih); last_out = prog.insert_instruction(ins, op::unsqueeze{{0, 1}}, sih);
std::size_t seq_len = input->get_shape().lens()[0]; std::size_t seq_len = input->get_shape().lens()[0];
for(std::size_t i = 0; i < seq_len; i++) for(std::size_t i = 0; i < seq_len; i++)
{ {
......
...@@ -1221,9 +1221,9 @@ struct test_rnn_3args ...@@ -1221,9 +1221,9 @@ struct test_rnn_3args
migraphx::shape w_shape{migraphx::shape::float_type, {num_dirct, hidden_size, input_size}}; migraphx::shape w_shape{migraphx::shape::float_type, {num_dirct, hidden_size, input_size}};
migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}}; migraphx::shape r_shape{migraphx::shape::float_type, {num_dirct, hidden_size, hidden_size}};
auto seq = p.add_parameter("seq", in_shape); auto seq = p.add_parameter("seq", in_shape);
auto w = p.add_parameter("w", w_shape); auto w = p.add_parameter("w", w_shape);
auto r = p.add_parameter("r", r_shape); auto r = p.add_parameter("r", r_shape);
p.add_instruction(migraphx::op::rnn{hidden_size, p.add_instruction(migraphx::op::rnn{hidden_size,
{migraphx::op::tanh{}, migraphx::op::tanh{}}, {migraphx::op::tanh{}, migraphx::op::tanh{}},
...@@ -1457,5 +1457,4 @@ int main() ...@@ -1457,5 +1457,4 @@ int main()
verify_program<test_rnn_5args>(); verify_program<test_rnn_5args>();
verify_program<test_rnn_bidirectional>(); verify_program<test_rnn_bidirectional>();
verify_program<test_rnn_bidirectional10>(); verify_program<test_rnn_bidirectional10>();
} }
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