Skip to content

GitLab

Commit 6c70c8a7 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 79081c0a
Hide whitespace changes
Inline Side-by-side
Showing with 105 additions and 105 deletions
test/onnx/onnx_rnn_test.cpp
View file @ 6c70c8a7
...@@ -1062,10 +1062,10 @@ TEST_CASE(lstm_bidirectional) ...@@ -1062,10 +1062,10 @@ TEST_CASE(lstm_bidirectional)
// 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(
migraphx::op::lstm{ migraphx::op::lstm{
...@@ -1092,11 +1092,11 @@ TEST_CASE(lstm_bidirectional) ...@@ -1092,11 +1092,11 @@ TEST_CASE(lstm_bidirectional)
// 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{
...@@ -1239,18 +1239,14 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1239,18 +1239,14 @@ TEST_CASE(lstm_bi_actv_funcs)
// 0 activation function // 0 activation function
{ {
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, hs, {}, migraphx::op::rnn_direction::bidirectional, clip, input_forget},
{},
migraphx::op::rnn_direction::bidirectional,
clip,
input_forget},
seq, seq,
w, w,
r, r,
...@@ -1269,27 +1265,26 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1269,27 +1265,26 @@ TEST_CASE(lstm_bi_actv_funcs)
// 1 activation function // 1 activation function
{ {
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 =
migraphx::op::lstm{ p.add_instruction(migraphx::op::lstm{hs,
hs, {migraphx::op::sigmoid{}},
{migraphx::op::sigmoid{}}, migraphx::op::rnn_direction::bidirectional,
migraphx::op::rnn_direction::bidirectional, clip,
clip, input_forget},
input_forget}, seq,
seq, w,
w, r,
r, bias,
bias, und,
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_bi1af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_bi1af.onnx");
...@@ -1307,21 +1302,20 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1307,21 +1302,20 @@ TEST_CASE(lstm_bi_actv_funcs)
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::sigmoid{}, migraphx::op::tanh{}},
{migraphx::op::sigmoid{}, migraphx::op::tanh{}}, migraphx::op::rnn_direction::bidirectional,
migraphx::op::rnn_direction::bidirectional, 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_bi2af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_bi2af.onnx");
...@@ -1340,21 +1334,23 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1340,21 +1334,23 @@ TEST_CASE(lstm_bi_actv_funcs)
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 =
migraphx::op::lstm{ p.add_instruction(migraphx::op::lstm{hs,
hs, {migraphx::op::sigmoid{},
{migraphx::op::sigmoid{}, migraphx::op::tanh{}, migraphx::op::tanh{}, migraphx::op::tanh{}}, migraphx::op::tanh{},
migraphx::op::rnn_direction::bidirectional, migraphx::op::tanh{},
clip, migraphx::op::tanh{}},
input_forget}, migraphx::op::rnn_direction::bidirectional,
seq, clip,
w, input_forget},
r, seq,
bias, w,
seq_len, r,
ih, bias,
und, seq_len,
und); ih,
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_bi4af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_bi4af.onnx");
...@@ -1374,22 +1370,24 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1374,22 +1370,24 @@ TEST_CASE(lstm_bi_actv_funcs)
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 =
migraphx::op::lstm{ p.add_instruction(migraphx::op::lstm{hs,
hs, {migraphx::op::sigmoid{},
{migraphx::op::sigmoid{}, migraphx::op::tanh{}, migraphx::op::tanh{}, migraphx::op::tanh{}, migraphx::op::tanh{},
migraphx::op::sigmoid{}}, migraphx::op::tanh{},
migraphx::op::rnn_direction::bidirectional, migraphx::op::tanh{},
clip, migraphx::op::sigmoid{}},
input_forget}, migraphx::op::rnn_direction::bidirectional,
seq, clip,
w, input_forget},
r, seq,
bias, w,
seq_len, r,
ih, bias,
ic, seq_len,
und); ih,
ic,
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_bi5af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_bi5af.onnx");
...@@ -1400,27 +1398,30 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1400,27 +1398,30 @@ TEST_CASE(lstm_bi_actv_funcs)
// 6 activation functions // 6 activation functions
{ {
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 =
migraphx::op::lstm{ p.add_instruction(migraphx::op::lstm{hs,
hs, {migraphx::op::sigmoid{},
{migraphx::op::sigmoid{}, migraphx::op::tanh{}, migraphx::op::tanh{}, migraphx::op::tanh{},
migraphx::op::tanh{}, migraphx::op::sigmoid{}, migraphx::op::tanh{}}, migraphx::op::tanh{},
migraphx::op::rnn_direction::bidirectional, migraphx::op::tanh{},
clip, migraphx::op::sigmoid{},
input_forget}, migraphx::op::tanh{}},
seq, migraphx::op::rnn_direction::bidirectional,
w, clip,
r, input_forget},
und, seq,
und, w,
und, r,
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_bi6af.onnx"); auto prog = migraphx::parse_onnx("onnx_lstm_bi6af.onnx");
...@@ -1430,4 +1431,3 @@ TEST_CASE(lstm_bi_actv_funcs) ...@@ -1430,4 +1431,3 @@ TEST_CASE(lstm_bi_actv_funcs)
} }
int main(int argc, const char* argv[]) { test::run(argc, argv); } int main(int argc, const char* argv[]) { test::run(argc, argv); }
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