Commit bad02557 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

refine the unit test

parent 7ac9c87b
...@@ -10,7 +10,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn> ...@@ -10,7 +10,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
{ {
migraphx::program p; migraphx::program p;
auto* mm = p.get_main_module(); auto* mm = p.get_main_module();
migraphx::shape s{migraphx::shape::half_type, {1, 5, 2, 2}}; migraphx::shape s{migraphx::shape::half_type, {1, 3, 224, 224}};
auto x = mm->add_parameter("x", s); auto x = mm->add_parameter("x", s);
std::vector<float> vec(s.elements(), 500.0f); std::vector<float> vec(s.elements(), 500.0f);
auto l = mm->add_literal(migraphx::literal(s, vec)); auto l = mm->add_literal(migraphx::literal(s, vec));
...@@ -18,7 +18,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn> ...@@ -18,7 +18,7 @@ struct test_mul_lrn : verify_program<test_mul_lrn>
auto y = mm->add_instruction(migraphx::make_op("relu"), xl); auto y = mm->add_instruction(migraphx::make_op("relu"), xl);
mm->add_instruction( mm->add_instruction(
migraphx::make_op("lrn", migraphx::make_op("lrn",
{{"alpha", 0.0001}, {"beta", 0.75}, {"bias", 1.0}, {"size", 5}}), {{"alpha", 0.0001}, {"beta", 0.75}, {"bias", 1.0}, {"size", 50}}),
y); y);
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