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

clang format

parent 78c3f579
...@@ -12,11 +12,12 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -12,11 +12,12 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
std::size_t seed = 1) std::size_t seed = 1)
{ {
migraphx::shape vars{migraphx::shape::float_type, {channels}}; migraphx::shape vars{migraphx::shape::float_type, {channels}};
auto scale = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + seed))); auto scale = mm->add_literal(migraphx::abs(migraphx::generate_literal(vars, 1 + seed)));
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
...@@ -24,11 +25,13 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -24,11 +25,13 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
migraphx::program p; migraphx::program p;
std::size_t ichannels = 64; std::size_t ichannels = 64;
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(
auto y = mm->add_parameter("y", {migraphx::shape::float_type, {1, ichannels, 56, 56}}); {migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 1));
auto v = mm->add_literal(migraphx::generate_literal({migraphx::shape::float_type, {ochannels, ichannels, 1, 1}}, 2)); 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 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);
...@@ -36,9 +39,9 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add> ...@@ -36,9 +39,9 @@ struct test_conv_bn_add : verify_program<test_conv_bn_add>
auto conv2 = mm->add_instruction(migraphx::make_op("convolution"), relu2, v); auto conv2 = mm->add_instruction(migraphx::make_op("convolution"), relu2, v);
auto bn2 = add_bn(mm, conv2, ochannels, 1); auto bn2 = add_bn(mm, conv2, ochannels, 1);
auto sum = mm->add_instruction(migraphx::make_op("add"), bn1, bn2); auto sum = mm->add_instruction(migraphx::make_op("add"), bn1, bn2);
auto r = mm->add_instruction(migraphx::make_op("relu"), sum); auto r = mm->add_instruction(migraphx::make_op("relu"), sum);
mm->add_return({r}); mm->add_return({r});
return p; return p;
} }
}; };
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