Commit 0b643932 authored by wsttiger's avatar wsttiger
Browse files

oops

parent 80db037c
...@@ -46,7 +46,8 @@ void fwd_conv_batchnorm_rewrite_test() ...@@ -46,7 +46,8 @@ void fwd_conv_batchnorm_rewrite_test()
auto variance = p.add_literal(migraph::literal{vars, {37.11f}}); auto variance = p.add_literal(migraph::literal{vars, {37.11f}});
p.add_instruction( p.add_instruction(
migraph::op::batch_norm_inference{}, conv, scale, bias, mean, variance); 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