Skip to content

GitLab

Commit d2e0502e authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 91c90830
Hide whitespace changes
Inline Side-by-side
Showing with 50 additions and 59 deletions
test/onnx/onnx_rnn_test.cpp
View file @ d2e0502e
...@@ -644,9 +644,9 @@ TEST_CASE(lstm_forward) ...@@ -644,9 +644,9 @@ TEST_CASE(lstm_forward)
// 3 args // 3 args
{ {
migraphx::program p; migraphx::program p;
auto seq = p.add_parameter("seq", seq_shape); auto seq = p.add_parameter("seq", seq_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);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
...@@ -674,11 +674,11 @@ TEST_CASE(lstm_forward) ...@@ -674,11 +674,11 @@ TEST_CASE(lstm_forward)
// 4 args // 4 args
{ {
migraphx::program p; migraphx::program p;
auto seq = p.add_parameter("seq", seq_shape); auto seq = p.add_parameter("seq", seq_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);
auto bias = p.add_parameter("bias", bias_shape); auto bias = p.add_parameter("bias", bias_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
migraphx::op::lstm{ migraphx::op::lstm{
...@@ -710,7 +710,7 @@ TEST_CASE(lstm_forward) ...@@ -710,7 +710,7 @@ TEST_CASE(lstm_forward)
auto r = p.add_parameter("r", r_shape); auto r = p.add_parameter("r", r_shape);
auto bias = p.add_parameter("bias", bias_shape); auto bias = p.add_parameter("bias", bias_shape);
auto seq_len = p.add_parameter("seq_len", sl_shape); auto seq_len = p.add_parameter("seq_len", sl_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
migraphx::op::lstm{ migraphx::op::lstm{
...@@ -743,7 +743,7 @@ TEST_CASE(lstm_forward) ...@@ -743,7 +743,7 @@ TEST_CASE(lstm_forward)
auto bias = p.add_parameter("bias", bias_shape); auto bias = p.add_parameter("bias", bias_shape);
auto seq_len = p.add_parameter("seq_len", sl_shape); auto seq_len = p.add_parameter("seq_len", sl_shape);
auto ih = p.add_parameter("h0", ih_shape); auto ih = p.add_parameter("h0", ih_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
migraphx::op::lstm{ migraphx::op::lstm{
...@@ -777,7 +777,7 @@ TEST_CASE(lstm_forward) ...@@ -777,7 +777,7 @@ TEST_CASE(lstm_forward)
auto seq_len = p.add_parameter("seq_len", sl_shape); auto seq_len = p.add_parameter("seq_len", sl_shape);
auto ih = p.add_parameter("h0", ih_shape); auto ih = p.add_parameter("h0", ih_shape);
auto ic = p.add_parameter("c0", ih_shape); auto ic = p.add_parameter("c0", ih_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
migraphx::op::lstm{ migraphx::op::lstm{
...@@ -820,18 +820,13 @@ TEST_CASE(lstm_forward_actv_func) ...@@ -820,18 +820,13 @@ TEST_CASE(lstm_forward_actv_func)
// no activation function specified // no activation function specified
{ {
migraphx::program p; migraphx::program p;
auto seq = p.add_parameter("seq", seq_shape); auto seq = p.add_parameter("seq", seq_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);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(
migraphx::op::lstm{ migraphx::op::lstm{hs, {}, migraphx::op::rnn_direction::forward, clip, input_forget},
hs,
{},
migraphx::op::rnn_direction::forward,
clip,
input_forget},
seq, seq,
w, w,
r, r,
...@@ -850,27 +845,25 @@ TEST_CASE(lstm_forward_actv_func) ...@@ -850,27 +845,25 @@ TEST_CASE(lstm_forward_actv_func)
// 1 activation function specified // 1 activation function specified
{ {
migraphx::program p; migraphx::program p;
auto seq = p.add_parameter("seq", seq_shape); auto seq = p.add_parameter("seq", seq_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);
auto bias = p.add_parameter("bias", bias_shape); auto bias = p.add_parameter("bias", bias_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs = p.add_instruction(migraphx::op::lstm{hs,
migraphx::op::lstm{ {migraphx::op::sigmoid{}},
hs, migraphx::op::rnn_direction::forward,
{migraphx::op::sigmoid{}}, clip,
migraphx::op::rnn_direction::forward, input_forget},
clip, seq,
input_forget}, w,
seq, r,
w, bias,
r, und,
bias, und,
und, und,
und, und);
und,
und);
p.add_instruction(migraphx::op::rnn_last_output{}, out_hs); p.add_instruction(migraphx::op::rnn_last_output{}, out_hs);
p.add_instruction(migraphx::op::lstm_last_cell_output{}, out_hs); p.add_instruction(migraphx::op::lstm_last_cell_output{}, out_hs);
auto prog = migraphx::parse_onnx("onnx_lstm_f1af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_f1af.onnx");
...@@ -885,24 +878,23 @@ TEST_CASE(lstm_forward_actv_func) ...@@ -885,24 +878,23 @@ TEST_CASE(lstm_forward_actv_func)
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);
auto bias = p.add_parameter("bias", bias_shape); auto bias = p.add_parameter("bias", bias_shape);
auto seq_len = p.add_parameter("seq_len", sl_shape); auto seq_len = p.add_parameter("seq_len", sl_shape);
auto und = p.add_instruction(migraphx::op::undefined{}); auto und = p.add_instruction(migraphx::op::undefined{});
auto out_hs = p.add_instruction( auto out_hs =
migraphx::op::lstm{ p.add_instruction(migraphx::op::lstm{hs,
hs, {migraphx::op::tanh{}, migraphx::op::sigmoid{}},
{migraphx::op::tanh{}, migraphx::op::sigmoid{}}, migraphx::op::rnn_direction::forward,
migraphx::op::rnn_direction::forward, clip,
clip, input_forget},
input_forget}, seq,
seq, w,
w, r,
r, bias,
bias, seq_len,
seq_len, und,
und, und,
und, und);
und);
p.add_instruction(migraphx::op::rnn_last_output{}, out_hs); p.add_instruction(migraphx::op::rnn_last_output{}, out_hs);
p.add_instruction(migraphx::op::lstm_last_cell_output{}, out_hs); p.add_instruction(migraphx::op::lstm_last_cell_output{}, out_hs);
auto prog = migraphx::parse_onnx("onnx_lstm_f2af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_f2af.onnx");
...@@ -911,7 +903,6 @@ TEST_CASE(lstm_forward_actv_func) ...@@ -911,7 +903,6 @@ TEST_CASE(lstm_forward_actv_func)
} }
} }
TEST_CASE(lstm_reverse) TEST_CASE(lstm_reverse)
{ {
std::size_t sl = 5; // sequence len std::size_t sl = 5; // sequence len
......
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