"vscode:/vscode.git/clone" did not exist on "cff1144da0f0c2b835cf9fe623c41b1004a599ed"
Commit eef9de6f authored by Paul's avatar Paul
Browse files

Formatting

parent ea87f25e
......@@ -122,7 +122,9 @@ migraph::argument run_gpu()
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) {
if(not migraph::verify_range(cpu, gpu))
......@@ -391,7 +393,8 @@ struct test_conv_bn_relu_pooling
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 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);
p.add_instruction(migraph::pooling{"average", {1, 1}, {2, 2}, {3, 3}}, relu);
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