Commit 5564172e authored by Shucai Xiao's avatar Shucai Xiao
Browse files

clang format

parent 78c3f579
...@@ -16,7 +16,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -16,7 +16,8 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
auto bias = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 2 + seed))); auto bias = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 2 + seed)));
auto mean = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 3 + seed))); auto mean = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 3 + seed)));
auto variance = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 4 + seed))); auto variance = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 4 + seed)));
return mm->add_instruction(migraphx::make_op("batch_norm_inference"), x, scale, bias, mean, variance); return mm->add_instruction(
migraphx::make_op("batch_norm_inference"), x, scale, bias, mean, variance);
} }
migraphx::program create_program() const migraphx::program create_program() const
...@@ -26,9 +27,11 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -26,9 +27,11 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
std::size_t ochannels = 256; std::size_t ochannels = 256;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
auto x = mm->add_parameter("x", {migraphx::shape::float_type, {1, ichannels, 56, 56}}); auto x = mm->add_parameter("x", {migraphx::shape::float_type, {1, ichannels, 56, 56}});
auto w = mm->add_literal(migraphx::generate_literal({migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1)); auto w = mm->add_literal(migraphx::generate_literal(
{migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1));
auto y = mm->add_parameter("y", {migraphx::shape::float_type, {1, ichannels, 56, 56}}); auto y = mm->add_parameter("y", {migraphx::shape::float_type, {1, ichannels, 56, 56}});
auto v = mm->add_literal(migraphx::generate_literal({migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 2)); auto v = mm->add_literal(migraphx::generate_literal(
{migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 2));
auto relu1 = mm->add_instruction(migraphx::make_op("relu"), x); auto relu1 = mm->add_instruction(migraphx::make_op("relu"), x);
auto conv1 = mm->add_instruction(migraphx::make_op("convolution"), relu1, w); auto conv1 = mm->add_instruction(migraphx::make_op("convolution"), relu1, w);
auto bn1 = add_bn(mm, conv1, ochannels, 1); auto bn1 = add_bn(mm, conv1, ochannels, 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