Skip to content

GitLab

Commit 60fb4d7f authored by Paul's avatar Paul
Browse files

Formatting

parent b639892a
Show whitespace changes
Inline Side-by-side
Showing with 16 additions and 15 deletions
test/gpu/miopen.cpp
View file @ 60fb4d7f
...@@ -441,13 +441,14 @@ struct test_conv_bn_relu_pooling ...@@ -441,13 +441,14 @@ struct test_conv_bn_relu_pooling
struct test_conv_bn_relu_pooling2 struct test_conv_bn_relu_pooling2
{ {
static migraph::instruction_ref add_bn(migraph::program& p, migraph::instruction_ref x, std::size_t channels) static migraph::instruction_ref
add_bn(migraph::program& p, migraph::instruction_ref x, std::size_t channels)
{ {
migraph::shape vars{migraph::shape::float_type, {channels}}; migraph::shape vars{migraph::shape::float_type, {channels}};
auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1+channels))); auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1 + channels)));
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::batch_norm_inference{}, x, scale, bias, mean, variance); return p.add_instruction(migraph::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