"megatron/data/vscode:/vscode.git/clone" did not exist on "f66c58a9b28e4b6f403a0beb782ff896fe34c8df"
Commit 611bc847 authored by Paul's avatar Paul
Browse files

Formatting

parent a9212aad
...@@ -386,9 +386,9 @@ struct test_conv_bn_relu_pooling ...@@ -386,9 +386,9 @@ struct test_conv_bn_relu_pooling
migraph::shape xs{migraph::shape::float_type, {1, 3, 224, 224}}; migraph::shape xs{migraph::shape::float_type, {1, 3, 224, 224}};
migraph::shape ws{migraph::shape::float_type, {64, 3, 7, 7}}; migraph::shape ws{migraph::shape::float_type, {64, 3, 7, 7}};
migraph::shape vars{migraph::shape::float_type, {64}}; migraph::shape vars{migraph::shape::float_type, {64}};
auto x = p.add_parameter("x", xs); auto x = p.add_parameter("x", xs);
auto w = p.add_parameter("w", ws); auto w = p.add_parameter("w", ws);
auto conv = p.add_instruction(migraph::convolution{{3, 3}, {2, 2}, {1, 1}}, x, w); auto conv = p.add_instruction(migraph::convolution{{3, 3}, {2, 2}, {1, 1}}, x, w);
auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1))); auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1)));
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)));
......
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