Commit eef9de6f authored by Paul's avatar Paul
Browse files

Formatting

parent ea87f25e
...@@ -122,7 +122,9 @@ migraph::argument run_gpu() ...@@ -122,7 +122,9 @@ migraph::argument run_gpu()
return migraph::gpu::from_gpu(p.eval(m)); return migraph::gpu::from_gpu(p.eval(m));
} }
void verify_args(const std::string& name, const migraph::argument& cpu_arg, const migraph::argument& gpu_arg) void verify_args(const std::string& name,
const migraph::argument& cpu_arg,
const migraph::argument& gpu_arg)
{ {
visit_all(cpu_arg, gpu_arg)([&](auto cpu, auto gpu) { visit_all(cpu_arg, gpu_arg)([&](auto cpu, auto gpu) {
if(not migraph::verify_range(cpu, gpu)) if(not migraph::verify_range(cpu, gpu))
...@@ -391,7 +393,8 @@ struct test_conv_bn_relu_pooling ...@@ -391,7 +393,8 @@ struct test_conv_bn_relu_pooling
auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1))); auto variance = p.add_literal(migraph::abs(migraph::generate_literal(vars, 1)));
auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2))); auto scale = p.add_literal(migraph::abs(migraph::generate_literal(vars, 2)));
auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3))); auto bias = p.add_literal(migraph::abs(migraph::generate_literal(vars, 3)));
auto bn = p.add_instruction(migraph::batch_norm_inference{}, conv, mean, variance, scale, bias); auto bn =
p.add_instruction(migraph::batch_norm_inference{}, conv, mean, variance, scale, bias);
auto relu = p.add_instruction(migraph::activation{"relu"}, bn); auto relu = p.add_instruction(migraph::activation{"relu"}, bn);
p.add_instruction(migraph::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu); p.add_instruction(migraph::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu);
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