Commit e83034cf authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent 5c0a7807
...@@ -512,8 +512,8 @@ struct test_conv_bn_relu_pooling ...@@ -512,8 +512,8 @@ struct test_conv_bn_relu_pooling
auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2))); auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2)));
auto mean = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3))); auto mean = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3)));
auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 4))); auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 4)));
auto bn = auto bn = p.add_instruction(
p.add_instruction(migraph::op::batch_norm_inference{}, conv, scale, bias, mean, variance); migraph::op::batch_norm_inference{}, conv, scale, bias, mean, variance);
auto relu = p.add_instruction(migraph::op::activation{"relu"}, bn); auto relu = p.add_instruction(migraph::op::activation{"relu"}, bn);
p.add_instruction(migraph::op::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu); p.add_instruction(migraph::op::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu);
return p; return p;
...@@ -530,7 +530,8 @@ struct test_conv_bn_relu_pooling2 ...@@ -530,7 +530,8 @@ struct test_conv_bn_relu_pooling2
auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2 + channels))); auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2 + channels)));
auto mean = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3 + channels))); auto mean = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3 + channels)));
auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 4 + channels))); auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 4 + channels)));
return p.add_instruction(migraph::op::batch_norm_inference{}, x, scale, bias, mean, variance); return p.add_instruction(
migraph::op::batch_norm_inference{}, x, scale, bias, mean, variance);
} }
migraph::program create_program() const migraph::program create_program() const
{ {
......
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