Commit 3a038fbe authored by wsttiger's avatar wsttiger
Browse files

Formatting

parent 0b643932
...@@ -34,20 +34,18 @@ void fwd_conv_batchnorm_rewrite_test() ...@@ -34,20 +34,18 @@ void fwd_conv_batchnorm_rewrite_test()
migraph::shape ws{migraph::shape::float_type, {1, 3, 3, 3}}; migraph::shape ws{migraph::shape::float_type, {1, 3, 3, 3}};
migraph::shape vars{migraph::shape::float_type, {1}}; migraph::shape vars{migraph::shape::float_type, {1}};
auto create_program = auto create_program = [&]() {
[&]() { migraph::program p;
migraph::program p; auto x = p.add_literal(xs, xdata);
auto x = p.add_literal(xs, xdata); auto w = p.add_literal(ws, wdata);
auto w = p.add_literal(ws, wdata); auto conv = p.add_instruction(migraph::op::convolution{{0, 0}, {1, 1}, {1, 1}}, x, w);
auto conv = p.add_instruction(migraph::op::convolution{{0, 0}, {1, 1}, {1, 1}}, x, w); auto scale = p.add_literal(migraph::literal{vars, {3.0f}});
auto scale = p.add_literal(migraph::literal{vars, {3.0f}}); auto bias = p.add_literal(migraph::literal{vars, {8.1f}});
auto bias = p.add_literal(migraph::literal{vars, {8.1f}}); auto mean = p.add_literal(migraph::literal{vars, {4.0f}});
auto mean = p.add_literal(migraph::literal{vars, {4.0f}}); auto variance = p.add_literal(migraph::literal{vars, {37.11f}});
auto variance = p.add_literal(migraph::literal{vars, {37.11f}}); p.add_instruction(migraph::op::batch_norm_inference{}, conv, scale, bias, mean, variance);
p.add_instruction( return p;
migraph::op::batch_norm_inference{}, conv, scale, bias, mean, variance); };
return p;
};
migraph::program p1 = create_program(); migraph::program p1 = create_program();
migraph::program p2 = create_program(); migraph::program p2 = create_program();
......
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